温馨提示×

python中groupby函数如何使用

小亿
116
2023-09-20 03:24:46
栏目: 编程语言

在Python中,groupby函数是通过pandas库的DataFrame对象使用的。该函数用于根据一个或多个列对数据进行分组。

下面是groupby函数的基本用法:

import pandas as pd
# 创建一个DataFrame对象
data = {'Name': ['Tom', 'Nick', 'John', 'Tom', 'John'],
'Age': [20, 25, 30, 20, 30],
'City': ['Seattle', 'New York', 'Seattle', 'Seattle', 'New York']}
df = pd.DataFrame(data)
# 使用groupby函数进行分组
grouped = df.groupby('City')
# 对分组后的数据执行聚合操作
result = grouped.mean()
print(result)

运行以上代码,输出结果如下:

Age
City
New York     27.5
Seattle      23.3

在上面的例子中,我们首先创建了一个DataFrame对象,其中包含了姓名、年龄和城市的数据。然后我们使用groupby函数根据城市进行分组。最后,我们对分组后的数据执行了均值操作,得到了每个城市对应的平均年龄。

除了使用单个列进行分组之外,我们还可以使用多个列进行分组。例如,可以在groupby函数中传入一个由多个列名组成的列表。例如:

grouped = df.groupby(['City', 'Age'])

这样就会根据城市和年龄进行分组。

除了mean函数,groupby函数还可以与其他聚合函数一起使用,如sum、count、min、max等。

希望以上内容能够帮助到你!

0