如何根据列值使用绘图分配颜色值？

Question

我正在尝试使用plotly 绘制k 均值簇，但我在根据这些组分配颜色时遇到困难？我有以下数据框。

group：簇号

我用它来绘制散点图。

clustered.iplot(kind='scatter',x='value1',y='value2', colors = {'[clustered['group']==1]':'green', '[clustered['group']==0]':'yellow'},mode='markers',size=10)

这是错误的，因为它只会为颜色字典对象获取 True 和 false。我如何关联这些组值，以便点的颜色在图中显示不同。

Answer 1

如果您只有两个集群，则可以映射值：

clustered.iplot(kind='scatter', x='value1', y='value2', colors=clustered['group'].map({0:'yellow', 1:'green'}), mode='markers',size=10)