Computer >> Máy Tính >  >> Lập trình >> Python

Thêm đường viền tùy chỉnh vào các ô nhất định trong lô Matplotlib / Seaborn

Để thêm đường viền tùy chỉnh vào các ô nhất định trong lô Matplotlib / Seaborn.

Các bước

  • Đặt kích thước hình và điều chỉnh phần đệm giữa và xung quanh các ô phụ.
  • Tạo khung dữ liệu với một số cột.
  • Vẽ biểu đồ tập dữ liệu ma trận dưới dạng bản đồ nhiệt được phân cấp theo nhóm.
  • Nhận trục bản đồ nhiệt dưới dạng sắp xếp ô con.
  • Để thêm một đường viền tùy chỉnh vào các ô nhất định trong Matplotlib, chúng tôi có thể phức tạp hóa một biến, border_color.
  • Sử dụng màu viền tùy chỉnh, thêm một mảng hình chữ nhật trên các trục bản đồ nhiệt.
  • Để hiển thị hình này, hãy sử dụng show () phương pháp.

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()

Đầu ra

Thêm đường viền tùy chỉnh vào các ô nhất định trong lô Matplotlib / Seaborn