为组中的每个数据点分配颜色

时间:2019-12-31 02:31:53

标签: list matplotlib colors bigdata scatter-plot

我为每个组分配不同的颜色时遇到问题。 有两组样本不同(第1组= 252个样本,第2组= 412个样本)

在我的工作的第一部分中,我使用以下代码创建了颜色列表,并且一切都很好地用于数据的可视化。

color_list = ['red' if i=='group1' else 'green' for i in data.loc[:,'classes']]
pd.plotting.scatter_matrix(data.iloc[:, 2:10], c=color_list)                                  

对于我的工作的第二部分,我想使用4个具有不同样本大小的组,但是在定义4个组的颜色列表时遇到问题。 我尝试使用elif,但它给我一个错误。

您如何像上述代码那样为4个或更多具有不同样本大小的组定义颜色列表?

谢谢

0 个答案:

没有答案