Thêm viền tùy chỉnh cho ô cụ thể trong biểu đồ Matplotlib/Seaborn

Để thêm viền tùy chỉnh cho một số ô trong heatmap Matplotlib/Seaborn.

Các bước thực hiện

• Thiết lập kích thước hình và khoảng đệm.
• Tạo DataFrame với vài cột.
• Vẽ heatmap phân cấp bằng clustermap().
• Lấy axes heatmap.
• Khởi tạo biến border_color.
• Thêm patch Rectangle lên axes heatmap với viền tùy chỉnh.
• Hiển thị bằng show().

Ví dụ

import pandas as pd
from matplotlib import pyplot as plt
import seaborn as sns

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

df = pd.DataFrame({"col1": [1, 4, 2, 3, 5], "col2": [3, 4, 1, 5, 2]})
g = sns.clustermap(df, figsize=(7.50, 3.50))
ax = g.ax_heatmap

border_color = "yellow"
ax.add_patch(plt.Rectangle((1, 2), 2, 1, fill=False, edgecolor=border_color, lw=5))
plt.show()

Kết quả