カラーマップ作成時、範囲外の値となるメッシュを白で描画したい

Question

環境

OS: Mac
エディタ: atom
言語: Python

Numpy以外のモジュールは使い慣れていません。。

実現したいこと

FitファイルのデータをPythonで読み込み、matplotlibでカラーマップを作成したいです(今回用いているFitファイルは'a.fit'という名前です)。

該当のソースコード

import astropy.io.fits as iofits
import matplotlib.pyplot as plt
import numpy as np

list = iofits.open("a.fit") #a.fitは1×301×301の3次元配列(実質2次元)
hdu = list[0]
data = np.array([hdu.data])
header = hdu.header
re = data.reshape(1*301, 301) #1×301×301の3次元配列を、301×301の2次元配列に変換

plt.imshow(re, cmap='nipy_spectral', vmin=5e+12, vmax=2e+14)
plt.colorbar()
plt.savefig('a.png', dpi=200) #画像保存

実現したいこと

カラーマップは上のコードで作成できましたが、もし可能なら、
ある値よりも大きい(または小さい)メッシュについて、白色で描画したいと考えています。

plt.imshow(re, cmap='nipy_spectral', vmin=5e+12, vmax=2e+14)

今回、're'というデータの値を5e+12から2e+14の範囲についてカラーマップで描画しましたが、

カラーマップのパラメータをcmap='nipy_spectral'としたことで、nipy_spectralのデフォルトで範囲外の値となるメッシュが灰色で描画されてしまいます。
(カラーバーを見ると、2e+14以上は灰色になっています)

そこで、'cmap.set_under('white')'や'cmap.set_over('white')'を使って範囲外の値となるメッシュの色を灰色→白色に変えたいのですが、

name 'cmap' is not defined

と出てきて、上手くいきませんでした。

どこを訂正すればよいでしょうか、教えていただきたいです。

Answer 1

解決しました。cmapを定義すればいいのですね

import astropy.io.fits as iofits
import matplotlib.pyplot as plt
import numpy as np

list = iofits.open("a.fit")
hdu = list[0]
data = np.array([hdu.data])
header = hdu.header
re = data.reshape(1*301, 301) #1×301×301の3次元配列を、301×301の2次元配列に変換

cmap = plt.cm.nipy_spectral #cmapを定義
# cmap.set_under('white')
cmap.set_over('white')

plt.imshow(re, cmap=cmap, vmin=5e+12, vmax=2e+14)
# plt.imshow(re, origin='lower', cmap='nipy_spectral', vmin=1e+13, vmax=1e+14)
plt.colorbar()
plt.savefig('a.png', dpi=200)