import matplotlib.pyplot as plt
import WrightTools as wt
from WrightTools import datasets
p = datasets.wt5.v1p0p1_MoS2_TrEE_movie
data = wt.open(p)
col = data.split('d2', -100.)
fig, gs = wt.artists.create_figure(cols=[1,1])
for i, d in enumerate(col.values()):
    d = d.chop("w1=wm", "d2", at={"w2": (2, "eV")})[0]
    ax = plt.subplot(gs[i])
    ax.pcolor(d)
    ax.set_xlim(data.w1__e__wm.min(), data.w1__e__wm.max())
    ax.set_ylim(data.d2.min(), data.d2.max())
wt.artists.set_fig_labels(xlabel=data.w1__e__wm.label, ylabel=data.d2.label)
plt.show()