Computer >> คอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

วิธีระบายสีกล่อง Seaborn ตามชื่อคอลัมน์ DataFrame ใน Matplotlib


ในการลงสีบ็อกซ์พล็อต Seaborn ตามชื่อคอลัมน์ dataframe เราสามารถทำตามขั้นตอนต่อไปนี้ -

  • กำหนดขนาดรูปและปรับช่องว่างภายในระหว่างและรอบๆ แผนผังย่อย
  • สร้างดาต้าเฟรม Pandas ที่มีสองคอลัมน์ col1 และ col2 .
  • สร้างบ็อกซ์พล็อตด้วยการวางแนวนอน
  • รับกล่องศิลปิน
  • วนซ้ำกล่องและกำหนดสีหน้าของกล่อง
  • หากต้องการแสดงรูป ให้ใช้ show() วิธีการ

ตัวอย่าง

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

ผลลัพธ์

วิธีระบายสีกล่อง Seaborn ตามชื่อคอลัมน์ DataFrame ใน Matplotlib