python imshow彩色_如何在matplotlib中使用imshow将NaN值绘制为特殊颜色？

我正在尝试在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设置为使用特殊颜色绘制(例如：灰色或透明)？