如何将NaN值绘制为一种特殊颜色,并在matplotlib中显示?

内容来源于 Stack Overflow,并遵循CC BY-SA 3.0许可协议进行翻译与使用

  • 回答 (2)
  • 关注 (0)
  • 查看 (20)

我试图使用matplotlib中的imShow来绘制数据作为热图,但是其中一些值是nans。我想把南安画成一种特殊的颜色,在彩色地图上找不到。

例子:

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的颜色相同...。有没有一种优雅的方式,我可以设置一个特殊的颜色(如:灰色或透明)的NANS?

提问于
用户回答回答于

人力资源管理,看来我可以使用蒙面数组来完成以下操作:

masked_array = np.ma.array (a, mask=np.isnan(a))
cmap = matplotlib.cm.jet
cmap.set_bad('white',1.)
ax.imshow(masked_array, interpolation='nearest', cmap=cmap)

这应该就够了,尽管我仍然愿意接受建议。

用户回答回答于

对于新版本的Matplotlib,不再需要使用蒙面数组。

例如,让我们生成一个数组,其中每7个值都有一个NaN:

arr = np.arange(100, dtype=float).reshape(10, 10)
arr[~(arr % 7).astype(bool)] = np.nan

我们可以修改当前的颜色图,并用以下行绘制数组:

current_cmap = matplotlib.cm.get_cmap()
current_cmap.set_bad(color='red')
plt.imshow(arr)

扫码关注云+社区