使用 matplotlib 可视化聚类
我们已经讨论过，可视化对于评估形成的聚类并发现数据中的趋势非常重要。现在请使用 matplotlib 模块来可视化来自 Comic-Con 的客流量数据集。
数据保存在 pandas DataFrame comic_con 中。x_scaled 和 y_scaled 是某一时刻人群标准化 X 和 Y 坐标的列名。cluster_labels 存放聚类标签。一个链接对象存储在变量 distance_matrix 中。
本练习是课程的一部分
Python 中的聚类分析
练习说明
- 从
matplotlib模块导入pyplot类并命名为plt。 - 为两个聚类标签
1和2定义一个colors字典。 - 绘制散点图，并根据
colors字典为每个聚类设置颜色。
交互式实操练习
通过完成这段示例代码来试试这个练习。
# Import the pyplot class
____
# Define a colors dictionary for clusters
colors = {____:'red', ____:'blue'}
# Plot a scatter plot
comic_con.plot.scatter(x=____,
y=____,
c=comic_con['cluster_labels'].apply(____))
plt.show()