指定 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()

输出

更新于: 16-3-2021

319 次浏览

启动您的职业生涯

完成课程认证

开始
广告