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

matplotlibpythondata visualization更新于 2025/9/10 23:40:17

要创建对某个值进行特殊处理的 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()