gpt4 book ai didi

python - 列表中的 Pandas groupby 值

转载 作者:太空宇宙 更新时间:2023-11-03 13:25:50 25 4
gpt4 key购买 nike

我正在尝试从 pandas df 返回一个 groupby。我希望对输出值求和而不是merged。但以下合并适当的列表

import pandas as pd

d = ({
'Id' : [1,2,2,1],
'Val' : ['A','B','B','A'],
'Output' : [[1,2,3,4,5],[5,3,3,2,1],[6,7,8,9,1],[6,7,8,9,1]],
})

df = pd.DataFrame(data = d)

df = df.groupby(['Id','Val']).agg({'Output':'sum'}, axis = 1)

输出:

                                Output
Id Val
1 A [1, 2, 3, 4, 5, 6, 7, 8, 9, 1]
2 B [5, 3, 3, 2, 1, 6, 7, 8, 9, 1]

预期输出:

                                Output
Id Val
1 A [7,9,11,13,6]
2 B [11,10,11,11,2]

最佳答案

或者使用转换为 np.array 的单行代码:

df = df.groupby(['Id','Val']).apply(lambda x: x.Output.apply(np.array).sum())
print(df)

输出:

Id  Val
1 A [7, 9, 11, 13, 6]
2 B [11, 10, 11, 11, 2]
dtype: object

关于python - 列表中的 Pandas groupby 值,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55700087/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com