linefinder.analyze_data.classifications module

Tools for reading worldline data

@author: Zach Hafen @contact: zachary.h.hafen@gmail.com @status: Development

class linefinder.analyze_data.classifications.Classifications(data_dir, tag)

Bases: object

Loads and analyzes data created by classify.py

get_data(data_key, mask=None, slice_index=None)

Get the data from the data dictionary. Useful (over just accessing the array) when applying additional functions onto it.

Parameters:

• data_key (str) – Key for the relevant data.
• mask (np.array of bools) – What mask to apply to the data, if any
• slice_index (int) – If getting only a particular slice (for the two dimensional arrays like ‘is_wind), what slice?

Returns:

Array of the requested data

Return type:

data_arr (np.array)