【Python】进阶学习：pandas--groupby()用法详解

高斯小哥2024-03-08 22:29

📊【Python】进阶学习：pandas--groupby()用法详解

🌈 个人主页：高斯小哥

🔥 高质量专栏：Matplotlib之旅：零基础精通数据可视化、Python基础【高质量合集】、PyTorch零基础入门教程👈 希望得到您的订阅和支持~

💡 创作高质量博文(平均质量分92+)，分享更多关于深度学习、PyTorch、Python领域的优质内容！（希望得到您的关注~）

🌵文章目录🌵

  • [🤔 一、为什么需要groupby()？](#🤔 一、为什么需要groupby()？)
  • [🎯 二、groupby()的基本用法](#🎯 二、groupby()的基本用法)
  • [📈 三、聚合运算](#📈 三、聚合运算)
  • [🛠️ 四、高级用法与技巧](#🛠️ 四、高级用法与技巧)
    • [🔧 应用自定义函数](#🔧 应用自定义函数)
    • [🔄 数据转换](#🔄 数据转换)
    • [🔍 过滤数据](#🔍 过滤数据)
  • [🛠️ 五、实际案例应用](#🛠️ 五、实际案例应用)
  • [🎉 六、总结](#🎉 六、总结)
  • [🤝 七、期待与你共同进步](#🤝 七、期待与你共同进步)

👋 欢迎来到Python进阶学习之旅！今天，我们将深入探讨pandas库中非常强大的groupby()函数。groupby()函数在数据分析和数据清洗中发挥着关键作用，能够帮助我们轻松地对数据进行分组、聚合和转换。

🤔 一、为什么需要groupby()？

在处理大量数据时，我们经常需要按照某个或多个特征对数据进行分组 ，以便更好地理解数据的结构和关系。例如，我们可能希望按照年份、地区或产品类别对数据进行分组，并对每个组进行聚合运算，如求和、平均值、最大值等。这时，groupby()函数就显得非常有用。

🎯 二、groupby()的基本用法

首先，我们需要导入pandas库，并创建一个示例数据集。然后，我们可以使用groupby()函数按照指定的列对数据进行分组。

python 复制代码 
import numpy as np
import pandas as pd

# 创建一个简单的DataFrame
data = {
    'A': ['foo', 'bar', 'foo', 'bar', 'foo', 'bar', 'foo', 'foo'],
    'C': np.random.randn(8),
    'D': np.random.randn(8)
}
df = pd.DataFrame(data)

# 使用groupby按列'A'进行分组
grouped = df.groupby('A')


# 打印分组后的GroupBy对象
print(grouped)

输出：

less 复制代码 
<pandas.core.groupby.generic.DataFrameGroupBy object at 0x000002B2C070B8E0>

上述代码将按照列'A'的值对DataFrame进行分组，并返回一个GroupBy对象。我们可以进一步对这个对象进行聚合运算。

📈 三、聚合运算

GroupBy对象提供了多种聚合函数，如sum()、mean()、max()等。我们可以使用这些函数对每个组进行聚合运算。

python 复制代码 
import numpy as np
import pandas as pd

# 创建一个简单的DataFrame
data = {
    'A': ['foo', 'bar', 'foo', 'bar', 'foo', 'bar', 'foo', 'foo'],
    'C': np.random.randn(8),
    'D': np.random.randn(8)
}
df = pd.DataFrame(data)

# 使用groupby按列'A'进行分组
grouped = df.groupby('A')

# 打印分组后的对象
print(grouped)

# 计算每个组的平均值
mean_grouped = grouped.mean()
print(mean_grouped)

# 计算每个组的总和
sum_grouped = grouped.sum()
print(sum_grouped)

输出：

less 复制代码 
            C         D
A                      
bar  0.658173 -0.225388
foo  0.778100 -0.164148
           C         D
A                     
bar  1.97452 -0.676164
foo  3.89050 -0.820740

除了内置的聚合函数外，我们还可以使用agg()函数应用自定义的聚合函数。例如，我们可以计算每个组的标准差：

python 复制代码 
import numpy as np
import pandas as pd

# 创建一个简单的DataFrame
data = {
    'A': ['foo', 'bar', 'foo', 'bar', 'foo', 'bar', 'foo', 'foo'],
    'C': np.random.randn(8),
    'D': np.random.randn(8)
}
df = pd.DataFrame(data)

# 使用groupby按列'A'进行分组
grouped = df.groupby('A')

# 打印分组后的对象
print(grouped)

# 计算每个组的标准差
std_grouped = grouped.agg(np.std)
print(std_grouped)

输出：

less 复制代码 
<pandas.core.groupby.generic.DataFrameGroupBy object at 0x000002B2F480B880>
            C         D
A                      
bar  0.101229  0.274698
foo  0.996597  0.812362

🛠️ 四、高级用法与技巧

除了基本的分组和聚合操作外，groupby()还提供了许多高级功能，如应用自定义函数、转换数据等。

🔧 应用自定义函数

我们可以使用apply()方法应用自定义函数到每个组。例如，我们可以定义一个函数来计算每个组的最大值和最小值之差：

python 复制代码 
import numpy as np
import pandas as pd

# 创建一个简单的DataFrame
data = {
    'A': ['foo', 'bar', 'foo', 'bar', 'foo', 'bar', 'foo', 'foo'],
    'C': np.random.randn(8),
    'D': np.random.randn(8)
}
df = pd.DataFrame(data)

# 使用groupby按列'A'进行分组
grouped = df.groupby('A')

# 打印分组后的对象
print(grouped)

# 定义一个自定义函数，计算每个组的最大值和最小值之差
def range_diff(group):
    return group.max() - group.min()

# 使用apply()应用自定义函数
diff_grouped = grouped.apply(range_diff)
print(diff_grouped)

输出：

less 复制代码 
<pandas.core.groupby.generic.DataFrameGroupBy object at 0x000002ACBD83AA60>
            C         D
A                      
bar  2.497695  1.086924
foo  2.826518  2.063781

🔄 数据转换

groupby()还提供了transform()方法，用于将聚合运算的结果广播到原始数据的每一行。这在数据转换中非常有用。

python 复制代码 
import numpy as np
import pandas as pd

# 创建一个简单的DataFrame
data = {
    'A': ['foo', 'bar', 'foo', 'bar', 'foo', 'bar', 'foo', 'foo'],
    'C': np.random.randn(8),
    'D': np.random.randn(8)
}
df = pd.DataFrame(data)

# 使用groupby按列'A'进行分组
grouped = df.groupby('A')

# 打印分组后的对象
print(grouped)

# 使用transform()方法将每个组的平均值广播到原始数据的每一行
mean_transformed = grouped['C'].transform('mean')
print(mean_transformed)

# 将转换后的平均值添加到原始DataFrame中
df['C_mean'] = mean_transformed
print(df)

输出：

less 复制代码 
<pandas.core.groupby.generic.DataFrameGroupBy object at 0x00000188A56DA8E0>
0    0.344876
1   -1.358760
2    0.344876
3   -1.358760
4    0.344876
5   -1.358760
6    0.344876
7    0.344876
Name: C, dtype: float64
     A         C         D    C_mean
0  foo  0.783914 -1.027288  0.344876
1  bar -2.072893 -0.972087 -1.358760
2  foo  0.035637 -0.315908  0.344876
3  bar -1.953068  0.409697 -1.358760
4  foo  0.576048 -0.258289  0.344876
5  bar -0.050318 -1.115734 -1.358760
6  foo  0.093456  0.106227  0.344876
7  foo  0.235322  1.365150  0.344876

🔍 过滤数据

除了聚合和转换外，我们还可以使用filter()方法根据条件过滤出满足条件的组。

python 复制代码 
import numpy as np
import pandas as pd

# 创建一个简单的DataFrame
data = {
    'A': ['foo', 'bar', 'foo', 'bar', 'foo', 'bar', 'foo', 'foo'],
    'C': np.random.randn(8),
    'D': np.random.randn(8)
}
df = pd.DataFrame(data)

# 使用groupby按列'A'进行分组
grouped = df.groupby('A')

# 打印分组后的对象
print(grouped)

# 使用filter()方法过滤出满足条件的组（例如，组的大小大于3）
filtered_groups = grouped.filter(lambda x: len(x) > 3)
print(filtered_groups)

输出：

less 复制代码 
<pandas.core.groupby.generic.DataFrameGroupBy object at 0x0000015ADE2FA940>
     A         C         D
0  foo  1.967217  0.005976
2  foo  0.950149  0.098143
4  foo  0.568101  1.461587
6  foo -1.905337 -1.106591
7  foo -0.168686  0.692850

🛠️ 五、实际案例应用

最后，让我们通过一个实际案例来演示如何应用groupby()函数进行数据分析和清洗。

假设我们有一个包含销售数据的DataFrame，其中包含日期、地区、产品名称、销售额等列。我们希望按地区和产品名称对数据进行分组，并计算每个组的总销售额。

python 复制代码 
import numpy as np
import pandas as pd

# 创建一个包含销售数据的DataFrame
sales_data = {
    'date': pd.date_range(start='2023-01-01', periods=100),
    'region': np.random.choice(['North', 'South', 'East', 'West'], size=100),
    'product': np.random.choice(['Product A', 'Product B', 'Product C'], size=100),
    'sales': np.random.rand(100) * 1000
}
df_sales = pd.DataFrame(sales_data)

# 按地区和产品名称对数据进行分组，并计算总销售额
grouped_sales = df_sales.groupby(['region', 'product'])['sales'].sum().reset_index()

# 打印分组后的销售额
print(grouped_sales)

输出：

less 复制代码 
   region    product        sales
0    East  Product A  2728.679432
1    East  Product B  1847.966730
2    East  Product C  4518.356763
3   North  Product A  5882.374531
4   North  Product B  5519.364196
5   North  Product C  4229.953852
6   South  Product A  5303.784425
7   South  Product B  2321.080682
8   South  Product C  4239.002167
9    West  Product A  1689.650513
10   West  Product B  4002.790867
11   West  Product C  4894.553548

在这个案例中，我们首先创建了一个包含销售数据的DataFrame。然后，我们使用groupby()函数按地区和产品名称对数据进行分组，并使用sum()函数计算每个组的总销售额。最后，我们使用reset_index()函数将结果转换为一个新的DataFrame，并打印出来。

🎉 六、总结

groupby()函数是pandas库中一个非常强大的工具，它允许我们按照一个或多个特征对数据进行分组，并对每个组进行聚合、转换和过滤操作。通过熟练掌握groupby()函数的用法，我们可以更高效地处理和分析大量数据，从而洞察数据的内在结构和关系。希望这篇博客能够帮助你更好地理解和应用groupby()函数！

🤝 七、期待与你共同进步

🌱 亲爱的读者，非常感谢你每一次的停留和阅读！你的支持是我们前行的最大动力！🙏

🌐 在这茫茫网海中，有你的关注，我们深感荣幸。你的每一次点赞👍、收藏🌟、评论💬和关注💖，都像是明灯一样照亮我们前行的道路，给予我们无比的鼓舞和力量。🌟

📚 我们会继续努力，为你呈现更多精彩和有深度的内容。同时，我们非常欢迎你在评论区留下你的宝贵意见和建议，让我们共同进步，共同成长！💬

💪 无论你在编程的道路上遇到什么困难，都希望你能坚持下去，因为每一次的挫折都是通往成功的必经之路。我们期待与你一起书写编程的精彩篇章！ 🎉

🌈 最后，再次感谢你的厚爱与支持！愿你在编程的道路上越走越远，收获满满的成就和喜悦！祝你编程愉快！🎉

相关推荐
L_cl2 分钟前
Python学习从0到1 day28 Python 高阶技巧 ⑧ 递归
学习
明月清风徐徐13 分钟前
Scrapy爬取豆瓣电影Top250排行榜
python·selenium·scrapy
theLuckyLong14 分钟前
SpringBoot后端解决跨域问题
spring boot·后端·python
Yongqiang Cheng17 分钟前
Python operator.itemgetter(item) and operator.itemgetter(*items)
python·operator·itemgetter
MavenTalk20 分钟前
Move开发语言在区块链的开发与应用
开发语言·python·rust·区块链·solidity·move
vortex520 分钟前
Vim 编辑器学习笔记
学习·编辑器·vim
源于花海23 分钟前
论文学习（四） | 基于数据驱动的锂离子电池健康状态估计和剩余使用寿命预测
论文阅读·人工智能·学习·论文笔记
心怀梦想的咸鱼26 分钟前
Ue5 umg学习（一）
学习·ue5
楚疏笃27 分钟前
鸿蒙学习生态应用开发能力全景图-开发者支持平台（5）
学习·华为·harmonyos
4v1d28 分钟前
边缘计算的学习
人工智能·学习·边缘计算
热门推荐
01玄机平台应急响应—webshell查杀02Ubuntu 20.04使用Livox mid 360 测试 FAST_LIO03【HarmonyOS】HUAWEI DevEco Studio 下载地址汇总04新版微信小程序获取用户手机号05文件或文件夹名称中有空格如何批量去除06组基轨迹建模 GBTM的介绍与实现（Stata 或 R）07【TC3xx芯片】TC3xx芯片电源管理系统PMS详解08优化手机性能，解决卡顿问题：关闭这3个微信开关，释放内存空间09基于YOLOv10深度学习的CT扫描图像肾结石智能检测系统【python源码+Pyqt5界面+数据集+训练代码】深度学习实战、目标检测10【Flutter】基于 GetX 实现 Dio 的生命周期自动化