python - Python 中的自定义颜色图

Question

我想在 python 中创建我自己的自定义颜色映射，我查看了一些在线示例并找到了命令

from matplotlib import cm 
import matplotlib.pyplot as plt 
from matplotlib.colors import LinearSegmentedColormap

cdict1 = {'red':   ((0.0, 0.0, 0.0),
                    (0.5, 0.0, 0.1),
                    (1.0, 1.0, 1.0)),

         'green': ((0.0, 0.0, 0.0),
                   (1.0, 0.0, 0.0)),

         'blue':  ((0.0, 0.0, 1.0),
                   (0.5, 0.1, 0.0),
                   (1.0, 0.0, 0.0))
         }

blue_red1 = LinearSegmentedColormap('BlueRed1', cdict1)
plt.imshow(big,interpolation='nearest',  cmap=blue_red1, aspect='auto')
plt.colorbar()
plt.show()

通过上面的命令，我得到了一个颜色图，它是(红-黑-蓝)，其中红色是最大的，蓝色是最小的。我想创建一个彩色 map (黑色 - 白色 - 黑色)。有人可以告诉我应该做什么或任何其他方法吗？

Answer 1

就其值(value)而言，还有一种更简单的方法。

LinearSegmentedColormap 的完整形式使您能够在同一颜色图中设置“硬”停止点和渐变，因此它必然很复杂。但是，对于您所描述的简单情况，有一个方便的构造函数。

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.colors import LinearSegmentedColormap

cmap = LinearSegmentedColormap.from_list('mycmap', ['black', 'white', 'black'])

fig, ax = plt.subplots()
im = ax.imshow(np.random.random((10, 10)), cmap=cmap, interpolation='nearest')
fig.colorbar(im)
plt.show()