在 Matplotlib 中定义颜色图的中点
matplotlibserver side programmingprogramming更新于 2024/11/21 7:43:00
使用 plt.subplots(1, 1) 方法，我们可以创建图和轴。我们可以使用 fig.colorbar 在图形的中点处制作颜色条。
步骤
使用 mgrid() 方法，`nd_grid` 实例返回一个开放的多维"meshgrid"。
创建 Z1、Z2 和 Z 数据。
使用 subplots 方法创建 fig 和 ax 变量，其中默认 nrows 和 ncols 为 1，使用 subplots() 方法。
使用 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()
输出
