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

Question

我正在尝试使用plotly绘制k-means聚类图，但是我在根据它们的分组分配颜色方面遇到了问题？ 我有以下数据框。

组：簇编号

我在plotly的散点图中使用它。

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

它是错误的，因为颜色字典对象只会得到True和false。我该如何将这些组值关联起来，以便点的颜色在图中呈现不同的样式。

- M_S_N

Answer 1

您可以使用新的Plotly Express库(https://plotly.express)来代替使用cufflinks，以完成以下操作:

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

- nicolaskruchten

Answer 2

如果你只有两个集群，你可以映射这些值：

最初的回答：

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

- zhavorsa

---

它没有起作用。我遇到了这个错误ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all(). - M_S_N