晕渲

用阴影图展示一些常见的技巧。

晕渲示例

晕渲示例2

晕渲示例3

  1. import numpy as np
  2. import matplotlib.pyplot as plt
  3. from matplotlib.colors import LightSource, Normalize
  6. def display_colorbar():
  7.     """Display a correct numeric colorbar for a shaded plot."""
  8.     y, x = np.mgrid[-4:2:200j, -4:2:200j]
  9.     z = 10 * np.cos(x**2 + y**2)
  11.     cmap = plt.cm.copper
  12.     ls = LightSource(315, 45)
  13.     rgb = ls.shade(z, cmap)
  15.     fig, ax = plt.subplots()
  16.     ax.imshow(rgb, interpolation='bilinear')
  18.     # Use a proxy artist for the colorbar...
  19.     im = ax.imshow(z, cmap=cmap)
  20.     im.remove()
  21.     fig.colorbar(im)
  23.     ax.set_title('Using a colorbar with a shaded plot', size='x-large')
  26. def avoid_outliers():
  27.     """Use a custom norm to control the displayed z-range of a shaded plot."""
  28.     y, x = np.mgrid[-4:2:200j, -4:2:200j]
  29.     z = 10 * np.cos(x**2 + y**2)
  31.     # Add some outliers...
  32.     z[100, 105] = 2000
  33.     z[120, 110] = -9000
  35.     ls = LightSource(315, 45)
  36.     fig, (ax1, ax2) = plt.subplots(ncols=2, figsize=(8, 4.5))
  38.     rgb = ls.shade(z, plt.cm.copper)
  39.     ax1.imshow(rgb, interpolation='bilinear')
  40.     ax1.set_title('Full range of data')
  42.     rgb = ls.shade(z, plt.cm.copper, vmin=-10, vmax=10)
  43.     ax2.imshow(rgb, interpolation='bilinear')
  44.     ax2.set_title('Manually set range')
  46.     fig.suptitle('Avoiding Outliers in Shaded Plots', size='x-large')
  49. def shade_other_data():
  50.     """Demonstrates displaying different variables through shade and color."""
  51.     y, x = np.mgrid[-4:2:200j, -4:2:200j]
  52.     z1 = np.sin(x**2)  # Data to hillshade
  53.     z2 = np.cos(x**2 + y**2)  # Data to color
  55.     norm = Normalize(z2.min(), z2.max())
  56.     cmap = plt.cm.RdBu
  58.     ls = LightSource(315, 45)
  59.     rgb = ls.shade_rgb(cmap(norm(z2)), z1)
  61.     fig, ax = plt.subplots()
  62.     ax.imshow(rgb, interpolation='bilinear')
  63.     ax.set_title('Shade by one variable, color by another', size='x-large')
  65. display_colorbar()
  66. avoid_outliers()
  67. shade_other_data()
  68. plt.show()

下载这个示例