指定 Matplotlib 中的色彩映射的中间点
使用 plt.subplots(1, 1) 方法,我们可以创建图像和坐标轴。我们可以使用 fig.colorbar 在图片的中间位置制作颜色条。
步骤
使用 mgrid() 方法,`nd_grid` 实例返回一个开放的多维“网格”。
创建 Z1、Z2 和 Z 数据。
使用子图方法创建 fig 和 ax 变量,其中,默认的 nrows 和 ncols 为 1,使用子图() 方法。
使用 colorbar() 方法为 ScalarMappable 实例 *mappable* 创建一个颜色条。
使用 plt.show(),我们可以显示图像。
实例
import numpy as np import matplotlib.pyplot as plt import matplotlib.colors as colors N = 100 X, Y = np.mgrid[-3:3:complex(0, N), -2:2:complex(0, N)] Z1 = np.exp(-(X)**2 - (Y)**2) Z2 = np.exp(-(X * 10)**2 - (Y * 10)**2) Z = Z1 + 50 * Z2 fig, ax = plt.subplots(1, 1) fig.colorbar(ax.pcolor(X, Y, Z, norm=colors.LogNorm(vmin=Z.min(), vmax=Z.max()), shading='auto')) plt.show()
输出
广告