我为每个组分配不同的颜色时遇到问题。 有两组样本不同(第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个或更多具有不同样本大小的组定义颜色列表?
谢谢