Python 导入在其他文件中定义的颜色映射

我有一个定义了一个颜色映射，我和imshow一起用于几个小脚本

colormap位于其自己的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)

现在我有一个不同的文件，其中有一个脚本，我想在其中绘制一个矩阵。我想要的是这样的代码：

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不是函数，所以我不能像这样导入它。那么有没有办法导入这样的颜色映射呢？

我找到了解决方案

使用

execfile（'colorcodes.py'）

将colorcode加载到我的新脚本中