在明显聚类上的肘部法
让我们使用 comic con 数据集,看看当数据具有清晰、分明的簇时,肘部图会是什么样子。开始练习前,您可以先把数据点可视化出来。
数据保存在一个 pandas DataFrame 中,名为 comic_con。x_scaled 和 y_scaled 是某一时刻人群位置的标准化 X、Y 坐标对应的列名。
本练习是课程的一部分
Python 中的聚类分析
交互式实操练习
通过完成这段示例代码来试试这个练习。
distortions = []
num_clusters = range(1, 7)
# Create a list of distortions from the kmeans function
for i in ____:
cluster_centers, distortion = ____
distortions.append(distortion)
# Create a DataFrame with two lists - num_clusters, distortions
elbow_plot = pd.DataFrame({'num_clusters': ____, 'distortions': ____})
# Creat a line plot of num_clusters and distortions
sns.lineplot(x=____, y=____, data = ____)
plt.xticks(num_clusters)
plt.show()