def plot_shaded(self, cmap=plt.cm.terrain, lightsource_kwargs=None, *args, **kwargs):
if lightsource_kwargs is None:
lightsource_kwargs = {'azdeg':225, 'altdeg':5}
extent = [self.x.min(), self.x.max(), self.y.min(), self.y.max()]
arr = self.z.copy()
nan_mask = np.isnan(arr)
arr_min = arr[~nan_mask].min()
if nan_mask.any():
arr[nan_mask] = max(arr_min-10, 0)
ls = LightSource(**lightsource_kwargs)
shaded = ls.shade(arr, cmap=cmap)
fig = plt.figure()
ax = fig.add_subplot(111)
im = ax.imshow(shaded, cmap=cmap, extent=extent, *args, **kwargs)
plt.colorbar(im)
ax.get_xaxis().get_major_formatter().set_useOffset(False)
ax.get_yaxis().get_major_formatter().set_useOffset(False)
plt.show()
def plot(self, cmap=plt.cm.terrain, *args, **kwargs):
fig = plt.figure()
ax = fig.add_subplot(111)
extent = [self.x.min(), self.x.max(), self.y.min(), self.y.max()]
ax.imshow(self.z, cmap=cmap, extent=extent, *args, **kwargs)
ax.get_xaxis().get_major_formatter().set_useOffset(False)
ax.get_yaxis().get_major_formatter().set_useOffset(False)
plt.show()
