在matplotlib中設置NaN值的顏色

Question

我正在嘗試將顏色繪制成一些數組，並將一些值轉換為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')                

有什么建議嗎？

Answer 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()