导入在不同文件中定义的颜色映射

Question

我定义了一个颜色映射表，用于在一些小脚本中配合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不是一个函数，所以我不能像这样导入它。那么，有没有办法导入这样的颜色映射表呢？

Answer 1

我找到了解决办法……

通过使用 execfile('colorcodes.py') 这个命令，我的新脚本就能加载颜色代码了……