Convenience data analysis functions

[AVB25]

Some convenience functions for data analysis that have been helpful to me and may be so to others. These are the same that I opened a pull request for a few months ago in oitg (OxfordIonTrapGroup/oitg#45) and then forgot about 😅

They parse the results dictionary yielded by load_results to translate all PYON into python objects and make data more accessible. There are also functions to sort a multi-dimensional scan, possibly with missing entries (only tested for 2-D scans so far).

Demo usage for loading and directly plotting a 2D scan with missing entries:

# Load the experiment data
results, axes, args, _ = load_ndscan(day=day, rid=rid, experiment=<experiment>)

# Get sorted data for two results channels
p_1 = results["measurement_1_p_0"].data
p_2 = results["measurement_1_p_1"].data

# Get sorted axis data for two axes and scale them
axis_0 = axes[0].data / axes[0].scale
axis_1 = axes[1].data / axes[1].scale

# Plot
plt.pcolormesh(axis_0, axis_1, p_1)

[AVB25]

Pushes were formatting to pass tests