Làm cách nào để tô màu một ô vuông Seaborn dựa trên tên cột DataFrame trong Matplotlib?

Để tô màu cho ô vuông Seaborn dựa trên tên cột khung dữ liệu, chúng ta có thể thực hiện các bước sau -

• Đặ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 Pandas có hai cột, col1 và col2 .
• Tạo biểu đồ hình hộp với hướng nằm ngang.
• Nhận các nghệ sĩ đóng hộp.
• Lặp lại các hộp và đặt màu mặt của hộp.
• Để hiển thị hình này, hãy sử dụng show () phương pháp.

Ví dụ

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

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

df = pd.DataFrame(
      [[2, 4],
      [7, 2]
      ], columns=['col1', 'col2'])

ax = sns.boxplot(data=df, orient='h')
boxes = ax.artists

for i, box in enumerate(boxes):
   if 'col1' in df.columns[i]:
      box.set_facecolor('r')
   else:
      box.set_facecolor('b')

plt.show()

Đầu ra