9
我有一个通过scikit-learn中的KMeans算法生成的图。群集对应于不同的颜色。这里是图, 使图例对应于matplotlib中散点的颜色
我需要这个图的图例对应于图中的丛集号。理想情况下,图例应该显示集群的颜色,标签应该是集群编号。谢谢。
编辑:我想我应该把一些代码,因为人们downvoting这
from sklearn.cluster import KMeans
km = KMeans(n_clusters=20, init='random')
km.fit(df) #df is the dataframe which contains points as coordinates
labels = km.labels_
plt.clf()
fig = plt.figure()
ax = fig.add_subplot(111, axisbg='w', frame_on=True)
fig.set_size_inches(18.5, 10.5)
# Plot the clusters on the map
# m is a basemap object
m.scatter(
[geom.x for geom in map_points],
[geom.y for geom in map_points],
20, marker='o', lw=.25,
c = labels.astype(float),
alpha =0.9, antialiased=True,
zorder=3)
m.fillcontinents(color='#555555')
plt.show()
你可以绘制每个集群有一个单独的'm.scatter()'命令,或作为替代化妆与所有类的离散的颜色条。在sklearn画廊举例说明如何做到这一点。人们无法运行你的例子,所以这是很难与... –
谢谢@ RutgerKassies ..我会检查出sklearn画廊 – Nitin
@Nitin,如果你想出了一个解决方案,请把它发回这里 – cd98