Як я можу побудувати значення NaN як особливий колір за допомогою imshow у matplotlib?

Я намагаюся використовувати imshow у matplotlib для побудови даних як теплової карти, але деякі значення - NaN. Я хотів би, щоб NaN відображались як особливий колір, якого немає в кольоровій карті.

приклад:

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

Отримане зображення несподівано стає блакитним (найнижчий колір у кольоровій карті). Однак, якщо я буду робити змову так:

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

- тоді я отримую щось краще, але значення NaN намальовані тим же кольором, що і vmin ... Чи є витончений спосіб встановити NaN для малювання спеціальним кольором (наприклад, сірим або прозорим)?

У нових версіях Matplotlib більше немає необхідності використовувати маскований масив.

Наприклад, давайте генеруємо масив, де кожне 7-е значення - NaN:

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

Ми можемо змінити поточну карту кольорів та побудувати графік масиву за допомогою таких рядків:

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

Hrm, здається, я можу використовувати замаскований масив для цього:

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)

Цього має бути достатньо, хоча я все ще відкритий для пропозицій. :]