从颜色列表构造matplotlib.colors.ListedColormap 相当简单。这是一个使用 tableau 20 调色板中前 4 种颜色的示例 -
import matplotlib.pyplot as plt
from matplotlib.colors import ListedColormap
from matplotlib import cm
# Tableau 20 color palette for demonstration
colors = [(31, 119, 180), (174, 199, 232), (255, 127, 14), (255, 187, 120)]
# Conversion to [0.0 - 1.0] from [0.0 - 255.0]
colors = [(e[0] / 255.0, e[1] / 255.0, e[2] / 255.0) for e in colors]
cmap = ListedColormap(colors)
a = np.outer(np.linspace(0, 1, 20), np.linspace(0, 1, 20))
im = plt.imshow(a, cmap=cmap)
plt.colorbar(im)
plt.show()
但是,如果您在颜色列表中还没有渐变(因为上面没有),那么改用 matplotlib.colors.LinearSegmentedColormap 可能会更有用。由于预期的格式,这有点棘手,
[...] segmentdata 参数是一个包含一组红色、绿色和蓝色条目的字典。每个条目应该是 x、y0、y1 元组的列表,在表中形成行 [...]。
表中给定颜色的每一行都是 x、y0、y1 元组的序列。在每个序列中,x 必须从 0 到 1 单调递增。对于介于 x[i] 和 x[i+1] 之间的任何输入值 z,给定颜色的输出值将在y1[i] 和y0[i+1] 之间进行线性插值
这样的字典可以通过下面例子中的方法算法生成
import matplotlib.pyplot as plt
from matplotlib.colors import LinearSegmentedColormap
from matplotlib import cm
# Tableau 20 color palette for demonstration
colors = [(31, 119, 180), (174, 199, 232), (255, 127, 14), (255, 187, 120)]
colors = [(e[0] / 255.0, e[1] / 255.0, e[2] / 255.0) for e in colors]
nc = len(colors)
c = np.zeros((3, nc, 3))
rgb = ['red', 'green', 'blue']
for idx, e in enumerate(colors):
for ii in range(3):
c[ii, idx, :] = [float(idx) / float(nc - 1), e[ii], e[ii]]
cdict = dict(zip(rgb, c))
cmap = LinearSegmentedColormap('tab20', cdict)
a = np.outer(np.linspace(0, 1, 20), np.linspace(0, 1, 20))
im = plt.imshow(a, cmap=cmap)
plt.colorbar(im)
plt.show()
假设输入列表 colors 具有正确的 RGB 格式。