플롯 색상 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()