如何在每组 pandas DataFrame 上应用聚合列表?
pythonserver side programmingprogramming更新于 2025/11/19 22:36:17
要应用聚合列表，请使用 agg() 方法。首先，导入所需的库 −
import pandas as pd
创建一个包含两列的 DataFrame −
dataFrame = pd.DataFrame(
{
"Car": ['BMW', 'Lexus', 'Lexus', 'Mustang', 'Bentley', 'Mustang'],"Units": [100, 150, 110, 80, 110, 90]
}
)
使用 agg() 指定列表作为参数 −
dataFrame = dataFrame.groupby('Car').agg(list)
示例
以下是完整代码 −
import pandas as pd
# 创建 DataFrame
dataFrame = pd.DataFrame(
{
"Car": ['BMW', 'Lexus', 'Lexus', 'Mustang', 'Bentley', 'Mustang'],"Units": [100, 150, 110, 80, 110, 90]
}
)
print("DataFrame ...\n",dataFrame)
# 使用 agg() 指定列表作为参数
dataFrame = dataFrame.groupby('Car').agg(list)
# 显示数据框列表
print("\nDataFrame ...\n",dataFrame)
输出
这将产生以下输出 −
DataFrame ... Car Units 0 BMW 100 1 Lexus 150 2 Lexus 110 3 Mustang 80 4 Bentley 110 5 Mustang 90 DataFrame ... Units Car BMW [100] Bentley [110] Lexus [150,110] Mustang [80,90]
相关文章
有用资源
python 参考教程 - 该教程包含有关 python 的更多信息:https://www.cainiaomax.com/python/
