通过附加列表在 Polars 中聚合

问题描述 投票:0回答:1

在 Python Polars 中,如何通过连接列表而不是创建嵌套列表来聚合?例如,我想在 id 上聚合这个数据框

id 名字
1 [“鲍勃”]
1 [“玛丽”,“苏”]

得到这个结果

id 名字
1 [“鲍勃”,“玛丽”,“苏”]

如果我使用

df.group_by('id').agg("name")
,我会得到一个嵌套列表,这是我不想要的:

id 名字
1 [[“鲍勃”],[“玛丽”,“苏”]]
python group-by python-polars
1个回答
0
投票

尝试在您的姓名栏上使用爆炸。

result_df = df.group_by('id').agg(pl.col('name').explode())
© www.soinside.com 2019 - 2024. All rights reserved.