向 plot 添加更多条

Question

我有一个 excel 文件，如下所示：

| Dept.               | Q1 Budget | Q1 Actual | Q2 Budget | Q2 Actual | Q3 Budget | Q3 Actual | Q4 Budget | Q4 Actual |
|---------------------|-----------|-----------|-----------|-----------|-----------|-----------|-----------|-----------|
| Distribution        | 470,000   | 476,300   | 476,300   | 482,000   | 400,000   | 470,000   | 610,000   | 408,000   |
| Facilities          | 676,300   | 674,000   | 800,500   | 847,000   | 763,000   | 714,000   | 763,000   | 770,000   |
| Human Resources     | 363,000   | 432,000   | 463,000   | 442,000   | 463,000   | 440,000   | 443,000   | 467,000   |
| Information Systems | 763,000   | 726,300   | 863,000   | 870,000   | 876,300   | 776,000   | 700,000   | 740,000   |

我想将 plot 与部门相邻的宿舍作为 label，将实际和预算作为分组条，而我希望最终得到的 plot 看起来像这样：

到目前为止，这是我的代码：

df = pd.read_excel(file)

cols = df['Dept.']
df = df.T
df.columns = cols
df = df.tail(-1)

x = np.arange(4)
width = 0.30

plt.bar(x-0.15, df.loc['Q1 Actual'], width, color='cornflowerblue')
plt.bar(x+0.15, df.loc['Q1 Budget'], width, color='cornflowerblue', alpha=0.2)
plt.xticks(x, cols, rotation=45)

这给了我第一季度想要的 plot，看起来像这样：

我 go 如何将我的 plot 扩展到每个季度？

Answer 1

您的数据是宽格式，如果您将数据融合为长格式，您会发现这样做更容易。 此外，seaborn 绘图库使使用分类变量为条形着色变得非常简单。

import seaborn as sns
import pandas as pd

# Make plot bigger
sns.set(rc={'figure.figsize':(16,9)})

df = pd.DataFrame({'Dept.':['Q1 Actual','Q1 Budget', 'Q2 Actual', 'Q2 Budget'],
                  'Distribution':[476300, 470000, 482000, 476300],
                  'Facilities':[674000, 676300, 847000, 800500],
                  'Human Resources': [432000, 363000, 442000, 463000],
                  'Information Systems': [726300, 763000, 870000, 863000]})

df = df.melt(id_vars='Dept.', var_name='Function', value_name='Expenditure')

p = df.loc[df['Dept.'].str.startswith('Q1')]
sns.barplot(data=p, x='Function',y='Expenditure', hue='Dept.');

或者你可以 plot 他们一次全部：

df['Quarter'] = df['Dept.'].str.split(' ').str[0]
g = sns.FacetGrid(df, col='Quarter', height=10)
g.map_dataframe(sns.barplot, x='Function', y='Expenditure', hue='Dept.')