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

Verileri ısı haritası olarak çizmek için matplotlib'de imshow'u kullanmaya çalışıyorum, ancak bazı değerler NaN'ler. NaN'lerin renk haritasında bulunmayan özel bir renk olarak görüntülenmesini istiyorum.

misal:

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

Ortaya çıkan görüntünün tamamı beklenmedik bir şekilde mavidir (jet renk haritasındaki en düşük renk). Ancak, komployu şu şekilde yaparsam:

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

- sonra daha iyi bir şey elde ederim, ancak NaN değerleri vmin ile aynı renkte çiziliyor ... NaN'leri özel bir renkle çizmek için zarif bir yol var mı (örneğin: gri veya şeffaf)?

Matplotlib'in daha yeni sürümleriyle, artık maskeli bir dizi kullanmak gerekli değildir.

Örneğin, her 7. değerin bir NaN olduğu bir dizi oluşturalım:

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

Mevcut renk haritasını değiştirebilir ve diziyi aşağıdaki satırlarla çizebiliriz:

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

Hrm, bunu yapmak için maskeli bir dizi kullanabilirim gibi görünüyor:

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 olmalı, yine de önerilere açık olsam da. :]