Trực quan hóa ma trận tương quan
Ma trận tương quan được tạo ở bài trước có thể được vẽ dưới dạng heatmap. Để làm vậy, bạn có thể dùng hàm heatmap() từ thư viện seaborn, hàm này có nhiều tham số để tùy chỉnh cách hiển thị của heatmap.
df_corr = df.corr()
sns.heatmap(df_corr)
plt.xticks(rotation=90)
plt.yticks(rotation=0)
Bạn có thể dùng các phương thức .xticks() và .yticks() để xoay nhãn trục, tránh bị chồng lấn.
Để tìm hiểu các tham số của hàm heatmap(), hãy xem trang này.
Bài tập này là một phần của khóa học
Trực quan hóa dữ liệu chuỗi thời gian trong Python
Hướng dẫn bài tập
- Import
seabornvới bí danhsns. - Tính tương quan giữa tất cả các cột trong DataFrame
meatbằng phương pháp Spearman và gán kết quả cho biến mới tên làcorr_meat. - Vẽ heatmap của
corr_meat.
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 seaborn library
import ____ as ____
# Get correlation matrix of the meat DataFrame: corr_meat
____ = ____.____(method=____)
# Customize the heatmap of the corr_meat correlation matrix
____(corr_meat,
annot=True,
linewidths=0.4,
annot_kws={"size": 10})
plt.xticks(rotation=90)
plt.yticks(rotation=0)
plt.show()