Pandas Datenrahmen Gruppierung Werte

Question

ich Pandas Datenrahmen wie diese haben,

dd = pd.DataFrame(
{'name': ['abc','bcd','abc'], 
'seconds': [75,77,90], 
}) 

Ich brauche die Sekunden Spalte in einer einzigen Liste für die Zeilen mit dem gleichen Namen zu kombinieren.

Ich bin in der Lage, dies mit for-Schleife zu tun,

names= list(set(dd['name'])) 
counter=[] 
for a in names: 
    counter.append(list(dd[dd['name'] == a]['seconds'])) 
end 
seconds_list = pd.DataFrame(
{'name': names, 
'seconds': counter, 
}) 

Ausgang:

Aber das braucht viel Zeit auf einem großen Datenrahmen. Eine einfache Möglichkeit, dies ohne eine for-Schleife zu erreichen?

Danke!

Answer 1

Verwenden groupby mit applylist:

df = dd.groupby('name')['seconds'].apply(list).reset_index() 
print (df) 

    name seconds 
0 abc [75, 90] 
1 bcd  [77] 

Answer 2

Verwenden groupby, agg und tolist:

dd.groupby('name')['seconds'].agg(lambda x: x.tolist()).reset_index(name='seconds') 

Output:

name seconds 
0 abc [75, 90] 
1 bcd  [77]