我试图返回groupby从pandas df。我希望不对输出值求和merged。但以下merges适当lists。
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
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句