开始使用免费开始使用

宝可梦目击:k-means 聚类

我们将继续上一练习中对传说宝可梦目击数据的调查。与上一练习相同,本题仍使用宝可梦目击的示例。在本练习中,您将使用 k-means 聚类对目击位置进行分组。

xy 是目击位置的 X、Y 坐标列,存放在 pandas 的 DataFrame df 中。您可以使用以下库:将 matplotlib.pyplot 导入为 pltseaborn 导入为 sns,以及 pandas 导入为 pd

本练习是课程的一部分

Python 中的聚类分析

查看课程

练习说明

  • 导入 kmeansvq 函数。
  • 使用 kmeans() 函数,在设定 2 个簇的情况下计算簇中心。
  • 使用 vq() 函数为每个数据点分配聚类标签。
  • 使用 seaborn 绘制散点图,并为每个簇指定不同的颜色。

交互式实操练习

通过完成这段示例代码来试试这个练习。

# Import kmeans and vq functions
from scipy.cluster.vq import ____, ____

# Compute cluster centers
centroids,_ = ____(____, ____)

# Assign cluster labels
df['cluster_labels'], _ = ____(____, ____)

# Plot the points with seaborn
sns.scatterplot(x=____, y=____, hue=____, data=df)
plt.show()
编辑并运行代码