Python: 1d kmeans cluster 数据点计数标签
Python: 1d kmeans cluster data point count label
我已经关注 并成功创建了绘图和分析。
我的问题：有没有办法添加一个标签来显示每个聚类中的值计数？
首先使用 pandas 获取簇大小
然后使用 matplotlib
将文本放在图上
cluster_sizes = df.groupby('label').size().reset_index(name='size')
cluster_size_0 = cluster_sizes[cluster_sizes['label'] == 0]['size'].item()
cluster_size_1 = cluster_sizes[cluster_sizes['label'] == 1]['size'].item()
cluster_centers = kmeans.cluster_centers_.ravel()
plt.text(cluster_centers[0], 0.4, cluster_size_0, fontsize=12)
plt.text(cluster_centers[1], 0.4, cluster_size_1, fontsize=12)
我已经关注
我的问题：有没有办法添加一个标签来显示每个聚类中的值计数？
首先使用 pandas 获取簇大小 然后使用 matplotlib
将文本放在图上cluster_sizes = df.groupby('label').size().reset_index(name='size')
cluster_size_0 = cluster_sizes[cluster_sizes['label'] == 0]['size'].item()
cluster_size_1 = cluster_sizes[cluster_sizes['label'] == 1]['size'].item()
cluster_centers = kmeans.cluster_centers_.ravel()
plt.text(cluster_centers[0], 0.4, cluster_size_0, fontsize=12)
plt.text(cluster_centers[1], 0.4, cluster_size_1, fontsize=12)