简单的轴线网格2

简单的轴线网格2

  1. import matplotlib.pyplot as plt
  2. from mpl_toolkits.axes_grid1 import ImageGrid
  3. def get_demo_image():
  4. import numpy as np
  5. from matplotlib.cbook import get_sample_data
  6. f = get_sample_data("axes_grid/bivariate_normal.npy", asfileobj=False)
  7. z = np.load(f)
  8. # z is a numpy array of 15x15
  9. return z, (-3, 4, -4, 3)
  10. F = plt.figure(1, (5.5, 3.5))
  11. grid = ImageGrid(F, 111, # similar to subplot(111)
  12. nrows_ncols=(1, 3),
  13. axes_pad=0.1,
  14. add_all=True,
  15. label_mode="L",
  16. )
  17. Z, extent = get_demo_image() # demo image
  18. im1 = Z
  19. im2 = Z[:, :10]
  20. im3 = Z[:, 10:]
  21. vmin, vmax = Z.min(), Z.max()
  22. for i, im in enumerate([im1, im2, im3]):
  23. ax = grid[i]
  24. ax.imshow(im, origin="lower", vmin=vmin,
  25. vmax=vmax, interpolation="nearest")
  26. plt.show()

下载这个示例