在 matplotlib 中为 NaN 值设置颜色答案

【问题标题】：Set color for NaN values in matplotlib在 matplotlib 中为 NaN 值设置颜色
【发布时间】：2016-12-12 12:52:01
【问题描述】：

我正在尝试绘制一些数组的颜色，并将一些值转换为 np.nan（以便于解释）并在绘制时期望不同的颜色（白色？），而不是导致绘图和颜色条出现问题。

#this is before converted to nan
array = np.random.rand(4,10)
plt.pcolor(array)
plt.colorbar(orientation='horizontal')

#conditional value converted to nan
array = np.random.rand(4,10)
array[array<0.5]=np.nan
plt.pcolor(array)
plt.colorbar(orientation='horizontal')

有什么建议吗？

【问题讨论】：

标签： python numpy matplotlib nan colorbar

【解决方案1】：

其中一个解决方案是绘制掩码数组，如下所示：

import matplotlib.pylab as plt
import numpy as np

#conditional value converted to nan
array = np.random.rand(4,10)
array[array<0.5]=np.nan
m = np.ma.masked_where(np.isnan(array),array)
plt.pcolor(m)
plt.colorbar(orientation='horizontal')       
plt.show()

【讨论】：