如何创建对某个值进行特殊处理的 matplotlib 颜色图?

matplotlibpythondata visualization

要创建对某个值进行特殊处理的 matplotlib 颜色图,我们可以采取以下步骤 −

  • 设置图形大小并调整子图之间和周围的填充。
  • 获取一个颜色图实例,名称为"rainbow"。
  • 使用 set_under('red') 方法设置超出范围的低值的颜色。
  • 使用 numpy 创建随机数据和 eps。
  • 创建一个图形和一组子图。
  • 使用 imshow() 方法将数据显示为图像,即在 2D 常规栅格上。
  • 创建一个ScalarMappable 实例的 colorbar,im
  • 要显示图形,请使用 show() 方法。

示例

import matplotlib.pyplot as plt
import numpy as np

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

cmap = plt.get_cmap('rainbow')
cmap.set_under('red')

data = np.random.randn(5, 5)
eps = np.spacing(0.0)

fig, ax = plt.subplots()
im = ax.imshow(data, interpolation='nearest', vmin=eps, cmap=cmap)
fig.colorbar(im, extend='min')

plt.show()

输出


相关文章