我正在尝试在 matplotlib 中使用 imshow 将数据绘制为热图,但其中一些值是 NaN。我希望将 NaN 呈现为颜色图中未找到的特殊颜色。
例子:
import numpy as np
import matplotlib.pyplot as plt
f = plt.figure()
ax = f.add_subplot(111)
a = np.arange(25).reshape((5,5)).astype(float)
a[3,:] = np.nan
ax.imshow(a, interpolation='nearest')
f.canvas.draw()
结果图像出乎意料地全是蓝色(喷射颜色图中的最低颜色)。但是,如果我像这样进行绘图:
ax.imshow(a, interpolation='nearest', vmin=0, vmax=24)
--然后我得到了更好的结果,但是 NaN 值被绘制成与 vmin 相同的颜色......有没有一种优雅的方式可以将 NaN 设置为使用特殊颜色(例如:灰色或透明)绘制?