如何转换scatter（来自matplotlib）的c变量的值？

我正在使用matplotlib Python库创建PCA图，我根据类值（它是0,1或2）选择每个点的颜色。要做到这一点，我使用名为c的参数：如何转换scatter（来自matplotlib）的c变量的值？

plt.scatter(pca_data[:, 0], pca_data[:, 1], c=[0,1,1,0,2])

我希望做的就是添加一个传奇，从情节的颜色与标签链接，但是，我找不到值怎么得（ 0,1或2）转换为实际颜色。

我曾经想过直接给颜色，但我希望这个过程是自动化的，所以无论实际的课程数量如何，它都可以工作。

我试过使用to_rgb (from matplotlib.colors)但由于值不在0和1之间，它不起作用，如果我缩放它们，最终会得到一个奇怪的颜色矢量（黑色，白色，黑色）。

有什么想法？

谢谢。

来源

2016-10-10 Ella N.

这里是有3个标签的例子（但该方法可扩展到任何数量的）：

import numpy as np 
import matplotlib.pyplot as plt 
import matplotlib.cm as cm 
import matplotlib.colors as colors 

pca_data = np.random.randn(100,2) #some fake data 
labels = np.random.randint(low=0,high=3,size=100) #some fake class labels 

cNorm = colors.Normalize(vmin=0,vmax=2) #normalise the colormap 
scalarMap = cm.ScalarMappable(norm=cNorm,cmap='hot') #map numbers to colors 

fig,ax = plt.subplots() 
for i in np.unique(labels): 
    ax.scatter(pca_data[labels==i,0],pca_data[labels==i,1],\ 
       c=scalarMap.to_rgba(i),label="class {}".format(i),s=100) 

ax.legend(scatterpoints=1)

来源

2016-10-10 16:26:02

如何转换scatter（来自matplotlib）的c变量的值？

回答

相关问题