[解決済み] matplotlibのimshowでNaN値を特別な色でプロットするには？

質問

matplotlibのimshowを使ってデータをヒートマップとしてプロットしようとしているのですが、値の一部が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)

<イグ