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

时间:2014-05-03 09:25:13

标签: python imshow

我有一个已定义的色彩映射,我将其与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)

嗯...因为FFT_colormap不是函数我不能像这样导入它。那么有没有办法导入这样的色彩图?

1 个答案:

答案 0 :(得分:-1)

我找到了解决方案......

带有execfile('colorcodes.py')

将colorcode加载到我的新脚本中......