如何在Matplotlib中基于DataFrame列名给Seaborn boxplot上色?
要根据数据帧列名为Seaborn boxplot上色,可以执行以下步骤:
- 设置图像大小,并调整子图之间和周围的填充。
- 创建一个带有两列 col1 和 col2 的Pandas数据帧。
- 创建一个水平方向的箱图。
- 获取箱子的艺术家。
- 迭代箱子并设置箱子的面颜色。
- 要显示图像,请使用 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()