python中groupby函数如何使用

39次阅读

共计 700 个字符，预计需要花费 2 分钟才能阅读完成。

在 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 函数中传入一个由多个列名组成的列表。例如：