Why some data is missing on a figure when I define the axes? My dataset is a DataArray and some values are masked

Question

I would like to be able to define my axis and not lose information. Currently, I am missing the last data point in both dimensions I defined (see figure).

DATA = xr.DataArray(np.random.rand(74,13),dims=('section','lag'))
DATA = DATA.where(DATA>.98,drop=True)

fig,(ax0,ax1) = plt.subplots(1,2,figsize=(10,2))

ax0.pcolor(DATA)
ax0.set(title='Axes not defined')

ax1.pcolor(DATA.lag, DATA.section, DATA)
ax1.set(title='Axes defined')

plt.show()

Answer 1

One way around this: Since my data is a DataArray, I could redefine the coordinates and plot it with xarray:

DATA.assign_coords(section=a,lag=b)
DATA.plot.pcolor()