Trực quan hóa các cụm với matplotlib
Chúng ta đã nói rằng trực quan hóa là cần thiết để đánh giá các cụm được hình thành và phát hiện xu hướng trong dữ liệu của bạn. Giờ hãy tập trung trực quan hóa bộ dữ liệu lưu lượng người (footfall) từ Comic-Con bằng mô-đun matplotlib.
Dữ liệu được lưu trong một pandas DataFrame, comic_con. x_scaled và y_scaled là tên các cột của tọa độ X và Y đã được chuẩn hóa của mọi người tại một thời điểm nhất định. cluster_labels chứa các nhãn cụm. Đối tượng liên kết (linkage) được lưu trong biến distance_matrix.
Bài tập này là một phần của khóa học
Phân cụm trong Python
Hướng dẫn bài tập
- Import lớp
pyplottừ mô-đunmatplotlibvới bí danhplt. - Định nghĩa một từ điển
colorscho hai nhãn cụm,1và2. - Vẽ biểu đồ scatter với màu cho từng cụm theo định nghĩa trong từ điển
colors.
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
# 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()