Я пытаюсь вернуть 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]