python - 将字符串与随机颜色匹配的优雅方法 matplotlib

Question

我想将一些数据的标签转换为颜色，以便使用 matplotlib 绘图

我有一个名字列表["bob", "joe", "andrew", "pete"]

在 matplotlib 中是否有一种内置的方法可以将这些字符串与颜色值进行映射？我考虑过随机创建十六进制值，但我最终可能会得到相似的颜色或不可见的颜色。

我尝试了几种不同的方法来尝试从下面的 cmap 答案中创建键值:

这个:

#names is a list of distinct names
cmap = plt.get_cmap('cool')
colors = cmap(np.linspace(0, 1, len(names)))
clr = {names[i]: colors[i] for i in range(len(names))}
ax.scatter(x, y, z, c=clr)

Answer 1

选择 color map , 例如 viridis:

cmap = plt.get_cmap('viridis')

颜色图 cmap 是一个函数，它可以获取从 0 到 1 的值数组并将它们映射到 RGBA 颜色。 np.linspace(0, 1, len(names)) 生成长度为 len(names) 的等间距数字数组，从 0 到 1。因此，

colors = cmap(np.linspace(0, 1, len(names)))

从 viridis 颜色图中选择等距颜色。

请注意，这不是使用字符串的值，它只是使用列表中字符串的序号位置来选择颜色。另请注意，这些不是随机颜色，这只是从任意字符串列表生成独特颜色的简单方法。

---

所以:

import numpy as np
import matplotlib.pyplot as plt

cmap = plt.get_cmap('viridis')
names = ["bob", "joe", "andrew", "pete"]
colors = cmap(np.linspace(0, 1, len(names)))
print(colors)
# [[ 0.267004  0.004874  0.329415  1.      ]
#  [ 0.190631  0.407061  0.556089  1.      ]
#  [ 0.20803   0.718701  0.472873  1.      ]
#  [ 0.993248  0.906157  0.143936  1.      ]]

x = np.linspace(0, np.pi*2, 100)
for i, (name, color) in enumerate(zip(names, colors), 1):
    plt.plot(x, np.sin(x)/i, label=name, c=color)
plt.legend()
plt.show()

---

问题

clr = {names[i]: colors[i] for i in range(len(names))}
ax.scatter(x, y, z, c=clr)

是 ax.scatter 的 c 参数需要一个 sequence RGB(A)与 x 相同长度的值或单一颜色。 clr 是字典，不是序列。所以如果 colors 与 x 的长度相同，那么您可以使用

ax.scatter(x, y, z, c=colors)