0

我有一个定义的颜色图，我将它与 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 不是函数，所以我不能像这样导入它。那么有没有办法导入这样的颜色图？

4

1 回答 1

-1

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

颜色代码已加载execfile('colorcodes.py')到我的新脚本中...

于 2014-05-03T09:38:41.167 回答