どのように私はmatplotlibの中imshowを持つ特殊な色としてNaN値をプロットすることができますか？

Question

私はヒートマップとしてプロットデータに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を設定できるという優雅な方法はありますか？

Answer 1

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)

私はまだ提案を開いてるのに

これは、十分です。 ：

Answer 2

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)