NaN değerlerini matplotlib'de imshow ile özel bir renk olarak nasıl çizebilirim?

Question

Verileri bir ısı haritası olarak çizmek için matplotlib'de imshow kullanmaya çalışıyorum, ancak bazı değerler NaN'dir. NaN'lerin renk eşleminde bulunmayan özel bir renk olarak gösterilmesini isterim.

örnek:

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

elde edilen görüntü tüm mavi (püskürtme renk haritası en düşük renk) beklenmedik bir şekilde. ... Ben NaN'ler ayarlayabilirsiniz zarif bir şekilde var mı daha iyi bir şey olsun -O zaman

ax.imshow(a, interpolation='nearest', vmin=0, vmax=24) 

ama NaN değerleri Vmin aynı renk çizilir: yapmam Ancak, böyle komplo özel bir renkle çizilmelidir (örneğin: gri veya şeffaf)?

Answer 1

Hrm, bunu yapmak maskeli dizi kullanabilirsiniz görüntülenir: Hala önerilere açığım olsa

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) 

Bu, yeterli olacaktır. :

Answer 2

Benim için çalışmadı. Ben hata mesajı başlamıştı, bu yüzden geçici çözümü yaptı: Matplotlib yeni sürümlerinde ile

a[3,:] = -999 
masked_array=np.ma.masked_where(a==-999, a) 
cmap = matplotlib.cm.jet 
cmap.set_bad('w',1.) 
ax.imshow(masked_array, interpolation='nearest', cmap=cmap) 

Answer 3

, artık bir maskeli dizi kullanmak gerekli değildir. her 7. değeri NaN burada Örneğin

, en bir dizi oluşturmak izin:

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

Biz mevcut renk haritası değiştirmek ve aşağıdaki çizgilerle dizi çizmek için:

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