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

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

您可以使用新的 Plotly Express 库 ( https://plotly.express ) 来代替使用 cufflinks 来执行此操作:

px.scatter(clustered, x='value1', y='value2', color='group')