色彩映射规范化边界

演示使用规范以非线性方式将颜色映射映射到数据上。

色彩映射规范化边界示例

  1. import numpy as np
  2. import matplotlib.pyplot as plt
  3. import matplotlib.colors as colors
  5. N = 100
  6. X, Y = np.mgrid[-3:3:complex(0, N), -2:2:complex(0, N)]
  7. Z1 = np.exp(-X**2 - Y**2)
  8. Z2 = np.exp(-(X - 1)**2 - (Y - 1)**2)
  9. Z = (Z1 - Z2) * 2
  11. '''
  12. BoundaryNorm: For this one you provide the boundaries for your colors,
  13. and the Norm puts the first color in between the first pair, the
  14. second color between the second pair, etc.
  15. '''
  17. fig, ax = plt.subplots(3, 1, figsize=(8, 8))
  18. ax = ax.flatten()
  19. # even bounds gives a contour-like effect
  20. bounds = np.linspace(-1, 1, 10)
  21. norm = colors.BoundaryNorm(boundaries=bounds, ncolors=256)
  22. pcm = ax[0].pcolormesh(X, Y, Z,
  23.                        norm=norm,
  24.                        cmap='RdBu_r')
  25. fig.colorbar(pcm, ax=ax[0], extend='both', orientation='vertical')
  27. # uneven bounds changes the colormapping:
  28. bounds = np.array([-0.25, -0.125, 0, 0.5, 1])
  29. norm = colors.BoundaryNorm(boundaries=bounds, ncolors=256)
  30. pcm = ax[1].pcolormesh(X, Y, Z, norm=norm, cmap='RdBu_r')
  31. fig.colorbar(pcm, ax=ax[1], extend='both', orientation='vertical')
  33. pcm = ax[2].pcolormesh(X, Y, Z, cmap='RdBu_r', vmin=-np.max(Z))
  34. fig.colorbar(pcm, ax=ax[2], extend='both', orientation='vertical')
  36. plt.show()

下载这个示例