散点图按颜色将python聚类

Question

我正在使用plotly（以便在悬停时能够获取点信息）来可视化我的聚集散点图。我在为使用KMeans生成的群集分配不同的颜色时遇到麻烦。当在matplotlib.pyplot中绘制此图形时（如plt），我使用以下代码：

plt.scatter(restult[:,0], result[:,1], c=cluster_labels

cluster_labels为：

n_clusters = 3
km = KMeans(n_clusters).fit(result)
labels = km.labels_

它工作正常，但是我需要鼠标悬停的信息。

这是我到目前为止所处的情节：

trace = go.Scatter(
    x = result[:,0],
    y = result[:,1],
    mode = 'markers',
    text = index, # I want to see the index of each point
)
data = [trace]

# Plot and embed in ipython notebook!
py.iplot(data, filename='basic-scatter')

感谢您的帮助！

Answer 1

• 让我们使用虹膜数据集
• 来自kmeans的标签用作颜色（marker=dict(color=kmeans.labels_)），就像在matplotlib中一样

from sklearn import datasets
from sklearn import cluster
import plotly
plotly.offline.init_notebook_mode()

iris = datasets.load_iris()
kmeans = cluster.KMeans(n_clusters=3, 
                        random_state=42).fit(iris.data[:,0:2])
data = [plotly.graph_objs.Scatter(x=iris.data[:,0], 
                                  y=iris.data[:,1], 
                                  mode='markers',     
                                  marker=dict(color=kmeans.labels_)
                                  )
       ]
plotly.offline.iplot(data)