如何在散点图上设置渐进色?

时间:2019-08-27 09:24:58

标签: python matplotlib

我试图用相应的标签表示2d空间中的点,我还希望这些点具有与其标签相同的颜色,最后我需要标签的颜色是渐进的。

我到处都是,但似乎找不到解决方法。 这是我当前的实现:

for i in range(0, encoder.classes_.size):
       plt.scatter(data_pca_reduced[i][0], data_pca_reduced[i][1],
                   label=i, cmap=plt.get_cmap('viridis'))

这就是结果:

Scatter plot result

您会看到标签的颜色不是渐进的。

1 个答案:

答案 0 :(得分:1)

感谢第二个链接正是我所寻找的!

现在这是工作代码:

colors = matplotlib.cm.rainbow(np.linspace(0, 1, encoder.classes_.size))
for i, c in zip(range(encoder.classes_.size), colors):
       plt.scatter(data_pca_reduced[i][0], data_pca_reduced[i][1], label=i, color=c)

这是结果: result image