
本教程旨在解决如何将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() 函数来绘制组合条形图。
基于VC与Matlab的混合编程实现图像的三维显示 WORD版
本文档主要讲述的是基于VC与Matlab的混合编程实现图像的三维显示;介绍了VC++与Matlab混合编程的一般实现方法,并实现对二维影像图的三维效果显示。 MATLAB既是一种直观、高效的计算机语言,同时又是一个科学计算平台。它为数据分析和数据可视化、算法和应用程序开发提供了最核心的数学和高级图形工具。希望本文档会给有需要的朋友带来帮助;感兴趣的朋友可以过来看看
9 查看详情
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/857445.html
微信扫一扫
支付宝扫一扫