groupby后reset_index的DataFrame如何绘制多X轴列等效柱状图

问题背景 #

你原本用这段pandas代码可以正常生成目标柱状图：

df.groupby(['State', 'City']).Total_sale_amount.sum().plot.bar()

之后通过以下代码得到了重置索引后的DataFrame：

df_city_sales = df.groupby(['City','State']).agg(Total_sale_amount=('Total_sale_amount', 'sum')
).reset_index()

但尝试多种plt.bar写法都无法生成等效图表，比如：

plt.bar(df_city_sales[['City', 'State']], df_city_sales['Total_sale_amount'])
plt.bar([df_city_sales['City'], df_city_sales['State']], df_city_sales['Total_sale_amount'])
plt.bar((df_city_sales['City'], df_city_sales['State']), df_city_sales['Total_sale_amount'])

原因说明 #

plt.bar的x轴参数要求是一维序列/数组，而你传入的是多列组合（二维DataFrame、多列列表/元组），matplotlib无法将其解析为有效的x轴标签，因此执行失败。

正确实现方法 #

需要把City和State合并成一个一维的字符串标签列，再传入plt.bar：

1. 先创建合并的标签列

df_city_sales['City_State'] = df_city_sales['City'] + ', ' + df_city_sales['State']

2. 使用合并后的列绘制柱状图

import matplotlib.pyplot as plt

plt.bar(df_city_sales['City_State'], df_city_sales['Total_sale_amount'])
# 可选：旋转x轴标签避免文字重叠，优化显示
plt.xticks(rotation=45, ha='right')
plt.tight_layout()
plt.show()

这样就能生成和原pandas代码等效的柱状图，每个柱子对应一个City+State的组合，x轴标签也和原图表保持一致。

内容的提问来源于stack exchange，提问作者gshep