本教程旨在解决如何将Pandas groupby()操作产生的不同聚合结果（如均值和总和）在同一张条形图中进行可视化的问题。通过详细讲解数据准备（使用reset_index()和pd.merge()合并数据框）和两种Matplotlib条形图绘制方法（垂直plt.bar和水平plt.barh），帮助读者高效地展示复杂的数据洞察。
在数据分析实践中，我们经常需要对数据进行分组聚合，并对比不同聚合指标（例如，某项指标的平均值与总和）在同一维度上的表现。直接使用Pandas groupby().plot.barh() 方法虽然方便，但它通常只能绘制单一聚合结果的图表。当需要将多个聚合结果（如均值和总和）组合到同一张图中时，我们需要更灵活的Matplotlib绘图技巧。
数据准备：聚合与合并
要将不同聚合结果组合到一张图表中，核心思路是先将这些聚合结果分别计算出来，然后将它们合并到一个统一的数据框中。这样，我们就可以基于这个合并后的数据框进行灵活的绘图。
执行分组聚合首先，对原始数据框 day_df 进行分组聚合。我们需要计算 cnt 列的均值 (mean) 和总和 (sum)。
import pandas as pdimport numpy as npimport matplotlib.pyplot as plt# 模拟数据，实际项目中请替换为您的day_dfdata = { 'yr': np.random.choice([0, 1], 100), 'season': np.random.choice([1, 2, 3, 4], 100), 'weathersit': np.random.choice([1, 2, 3], 100), 'cnt': np.random.randint(100, 1000, 100)}day_df = pd.DataFrame(data)# 计算均值day_mean_dataframe = day_df.groupby(by=["yr", "season", "weathersit"]).agg({"cnt": "mean"})# 计算总和day_sum_dataframe = day_df.groupby(by=["yr", "season", "weathersit"]).agg({"cnt": "sum"})
重置索引 (reset_index())groupby()操作通常会生成一个多级索引（MultiIndex）。为了方便后续的合并操作，我们需要将这些索引转换为普通列。reset_index()方法可以实现这一点。
day_mean_dataframe = day_mean_dataframe.reset_index()day_sum_dataframe = day_sum_dataframe.reset_index()
合并数据框 (pd.merge())现在我们有了两个数据框，一个包含均值，一个包含总和。使用 pd.merge() 函数将它们合并。关键是指定用于合并的共同列（on 参数）以及为区分相同列名（例如 cnt）而添加的后缀（suffixes 参数）。
merged_df = pd.merge(day_mean_dataframe, day_sum_dataframe, on=["yr", "season", "weathersit"], suffixes=('_mean', '_sum'))
合并后的 merged_df 将包含 yr, season, weathersit 作为共同的标识列，以及 cnt_mean 和 cnt_sum 两列，分别代表聚合的均值和总和。
立即学习“Python免费学习笔记（深入）”；
可视化实现：组合条形图
数据准备就绪后，我们可以使用Matplotlib的 plt.bar() 或 plt.barh() 函数来绘制组合条形图。
A. 垂直条形图 (plt.bar)
垂直条形图适用于类别标签较少的情况。我们将均值和总和的条形图并排绘制。
# 创建图和轴对象fig, ax = plt.subplots(figsize=(14, 7))# 为每个分组创建一个位置数组r = np.arange(len(merged_df))width = 0.4 # 条形图的宽度# 绘制均值条形图ax.bar(r - width/2, merged_df["cnt_mean"], width=width, label='平均值', color='skyblue')# 绘制总和条形图，稍微偏移以便并排显示ax.bar(r + width/2, merged_df["cnt_sum"], width=width, label='总和', color='lightcoral')# 设置X轴刻度标签# 将yr, season, weathersit组合成一个字符串作为标签ax.set_xticks(r)ax.set_xticklabels([f'{row.yr}, S{row.season}, W{row.weathersit}' for _, row in merged_df.iterrows()], rotation=90, ha='center') # 旋转标签以防重叠# 添加图例和轴标签ax.legend()ax.set_xlabel('年份, 季节, 天气状况')ax.set_ylabel('计数')ax.set_title('不同天气状况下共享单车计数（平均值与总和）')plt.tight_layout() # 自动调整布局，防止标签重叠plt.show()
B. 水平条形图 (plt.barh)
当类别标签较多或标签内容较长时，水平条形图 (plt.barh) 通常是更好的选择，因为它可以提供更多的空间来显示标签。
# 创建图和轴对象fig, ax = plt.subplots(figsize=(14, 8))# 为每个分组创建一个位置数组r = np.arange(len(merged_df))height = 0.4 # 条形图的高度# 绘制均值水平条形图ax.barh(r - height/2, merged_df["cnt_mean"], height=height, label='平均值', color='skyblue')# 绘制总和水平条形图，稍微偏移以便并排显示ax.barh(r + height/2, merged_df["cnt_sum"], height=height, label='总和', color='lightcoral')# 设置Y轴刻度标签# 将yr, season, weathersit组合成一个字符串作为标签ax.set_yticks(r)ax.set_yticklabels([f'{row.yr}, S{row.season}, W{row.weathersit}' for _, row in merged_df.iterrows()])# 添加图例和轴标签ax.legend()ax.set_xlabel('计数')ax.set_ylabel('年份, 季节, 天气状况')ax.set_title('不同天气状况下共享单车计数（平均值与总和）')plt.tight_layout() # 自动调整布局，防止标签重叠plt.show()
注意事项与最佳实践
索引处理：reset_index() 是将多级索引转换为列的关键一步，它使得 pd.merge() 能够正确地识别合并键，并简化后续的绘图操作。标签可读性：当分组维度较多时（如本例中的 yr, season, weathersit），将它们组合成一个有意义的字符串作为刻度标签非常重要。对于垂直条形图，可能需要 rotation=90 来防止标签重叠；对于水平条形图，标签通常有足够的空间。图表美化：始终添加清晰的图例 (ax.legend())、轴标签 (ax.set_xlabel(), ax.set_ylabel()) 和图表标题 (ax.set_title())，以提高图表的可读性和信息传达能力。条形图间距：通过调整 width (垂直条形图) 或 height (水平条形图) 以及条形图的起始位置 (r – width/2, r + width/2)，可以控制条形图之间的间距和排列方式。选择合适的图表类型：根据数据特征和可视化目标，选择垂直或水平条形图。当分类数量多或标签长时，水平条形图通常更优。
总结
通过以上步骤，我们学习了如何将Pandas groupby()产生的不同聚合结果有效地合并并可视化在同一张条形图中。关键在于：
分别计算所需的聚合指标。使用 reset_index() 将多级索引转换为列。利用 pd.merge() 将不同的聚合结果合并到一个统一的数据框中。使用Matplotlib的 plt.bar() 或 plt.barh() 函数，结合 np.arange() 和适当的偏移量，绘制并排的条形图。优化刻度标签、图例和轴标签，确保图表清晰易懂。
掌握这些技巧，将使您能够更灵活、更专业地展示复杂的数据分析结果。
以上就是Python数据可视化：组合多聚合结果的条形图的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1375656.html
微信扫一扫 
支付宝扫一扫 
