xarray pcolormesh and contour plotting of gridded 3d data

xarray-da-pcolormesh-example.py

import xarray as xr
import numpy as np
import matplotlib.pyplot as plt

# make fake data grid of x and y values
x, y, = np.meshgrid(np.arange(-5,5,0.01),np.arange(-5,5,0.01))
# compute some function on x and y to get z values
z = np.sin(x**2 + y**2) / (x**2 + y**2)

# create xr data array
da = xr.DataArray(data=z,
                  dims=["x", "y"],
                  coords=dict(
                      x_coord=(["x", "y"], x),
                      y_coord=(["x", "y"], y))
                 )

# make figure
fig, ax = plt.subplots(figsize=(7,6))
# pcolormesh
pcolormeshplot = da.plot.pcolormesh(x='x_coord', y='y_coord', ax=ax, add_colorbar=False)
# contour lines
countourplot = da.plot.contour(x='x_coord', y='y_coord', colors='k', levels=[0], ax=ax, linewidths=0.8, linestyles='dotted')
# colorbar
cbar = plt.colorbar(pcolormeshplot)
cbar.add_lines(countourplot)