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()