atomcloud.process_fits.plot module

class atomcloud.process_fits.plot.CloudFitPlots

Bases: IterateFitDict

Class to plot fits from fit dictionaries.

mixed_level_fit(all_fit_dicts, coords, data, mask, title, *args, **kwargs)

Process a mixed level fit dictionary. The mixed level fit is composed of either 1d multi-function fits, 2d multi-function fits, or sum fits. This function iterates through each fit level and calls the appropriate processing function for each fit level.

Parameters:

• all_fit_dicts – dictionary of fit dictionaries

• *args – additional arguments

• **kwargs – additional keyword arguments

Returns:

dictionary of processed fit dictionaries

multi_level_data(key, data)

plot_fitdict(fit_dicts, coords, data, dict_type=None, mask=None, title='', *args, **kwargs)

Parameters:

• fit_dicts (dict) –

• coords (ndarray | dict[str, Union[numpy.ndarray, Iterable[numpy.ndarray]]]) –

• data (ndarray | dict[str, numpy.ndarray]) –

• dict_type (str | None) –

• mask (ndarray | dict[str, numpy.ndarray] | None) –

• title (str) –

process_fitdict1d(fit_dicts, *args, **kwargs)

Plot the results from a single 1d fit dictionary.

process_fitdict2d(fit_dicts, *args, **kwargs)

Plot the results from a single 2d fit dictionary.

sum_fit(fit_dict, *args, **kwargs)

Plot the results from a single sum fit dictionary.

atomcloud.process_fits.plot.check_coords(coords, dict_type)

atomcloud.process_fits.plot.check_data(data, dict_type)

atomcloud.process_fits.plot.check_mask(mask, data, dict_type)

atomcloud.process_fits.plot.check_mixed_level_coords(coords)

atomcloud.process_fits.plot.plot_fitdict(fit_dicts, coords, data, mask=None, title='', dict_type=None, *args, **kwargs)

Parameters:

• fit_dicts (dict) –

• coords (ndarray | dict[str, Union[numpy.ndarray, Iterable[numpy.ndarray]]]) –

• data (ndarray | dict[str, numpy.ndarray]) –

• mask (ndarray | dict[str, numpy.ndarray] | None) –

• title (str) –

• dict_type (str | None) –