Level

Leveling a dataset.

Out:

axis w2 converted from nm to eV
axis w1=wm converted from nm to eV
chopped data into 1 piece(s) in ('w1=wm', 'd2')
channel ai0 leveled along axis 2
chopped data into 1 piece(s) in ('w1=wm', 'd2')

import matplotlib.pyplot as plt

import WrightTools as wt
from WrightTools import datasets

fig, gs = wt.artists.create_figure(width="double", cols=[1, 1, "cbar"])

p = datasets.wt5.v1p0p1_MoS2_TrEE_movie
data = wt.open(p)
data.convert("eV")
data.ai0.symmetric_root(2)

# as taken
ax = plt.subplot(gs[0, 0])
chop = data.chop("w1=wm", "d2", at={"w2": [1.7, "eV"]})[0]
chop.ai0.null = chop.ai0.min()  # only for example
ax.pcolor(chop)
ax.contour(chop)

# leveled
ax = plt.subplot(gs[0, 1])
data.level(0, 2, -3)
chop = data.chop("w1=wm", "d2", at={"w2": [1.7, "eV"]})[0]
chop.ai0.clip(min=0, replace="value")
ax.pcolor(chop, vmin=0)
ax.contour(chop)

# label
wt.artists.set_fig_labels(xlabel=data.w1__e__wm.label, ylabel=data.d2.label)

# colorbar
cax = plt.subplot(gs[0, -1])
wt.artists.plot_colorbar(cax=cax, label="amplitude")
wt.artists.set_ax_labels(cax, yticks=False)