いくつかの小さなスクリプトに imshow で使用する定義済みのカラーマップがあります...
カラーマップは、次のような独自の colormap.py ファイルにあります。
import matplotlib
#define colorcode for matrix
cdict = {'red': ((0,1,1),
(0.2,0,0),
(0.5,0.275,0.275),
(0.7,1,1),
(1,1,1)
),
'green': ((0,1,1),
(0.2,0.275,0.275),
(0.5,1,1),
(0.7,1,1),
(1,0,0)
),
'blue': ((0,1,1),
(0.2,1,1),
(0.5,0,0),
(0.7,0,0),
(1,0,0)
)}
FFT_colormap = matplotlib.colors.LinearSegmentedColormap('my_colormap',cdict,256)
これで、マトリックスをプロットするスクリプトを含む別のファイルができました。私が欲しいのは、次のように機能するコードです。
from colormap.py import FFT_colormap
import pylab as pl
pl.imshow(data, cmap=FFT_colormap)
うーん... FFT_colormap は関数ではないので、このようにインポートすることはできません。そのようなカラーマップをインポートする方法はありますか?