Python Pandas: 使用 groupby 将结果转换为 DataFrame
要将 groupby 的结果转换为 DataFrame,可以使用 'reset_index()' 方法。这将把 groupby 的结果重新设置为 DataFrame 的列。
下面是一个示例代码:
import pandas as pd
# 创建一个示例数据集
data = {
'Name': ['Tom', 'Nick', 'John', 'Tom', 'John'],
'Age': [20, 21, 22, 20, 22],
'City': ['New York', 'Paris', 'London', 'New York', 'London']
}
df = pd.DataFrame(data)
# 使用 groupby 对 Name 列进行分组,并计算每个分组的平均年龄
grouped = df.groupby('Name')['Age'].mean()
# 将 groupby 结果转换为 DataFrame
df_grouped = grouped.reset_index()
# 打印结果
print(df_grouped)
输出结果为:
Name Age
0 John 22.0
1 Nick 21.0
2 Tom 20.0
在上面的示例中,我们首先创建了一个示例数据集,并使用 'groupby' 对 'Name' 列进行分组,并计算每个分组的平均年龄。然后,使用 'reset_index()' 将 groupby 的结果转换为 DataFrame,最后打印结果。
原文地址: http://www.cveoy.top/t/topic/ii2H 著作权归作者所有。请勿转载和采集!