Pandas Datenrahmen auf mehrere Spalten und Werte dict zur Liste

Question

Ich habe einen Datenrahmen

id key 
a1  1 
a2  1 
a3  1 
a4  2 
a5  2 
a6  3 

ich nicht ein Wörterbuch mit key als Maschine erstellen möchten, und id Spalte als Liste

wie:

{1: ['a1', 'a2', 'a3'], 2: ['a4', 'a5'], 3: ['a6']} 

Kann ich zuerst groupby verwenden und dann .to_dict?

Answer 1

Ich glaube, Sie Listen Anzeige Werte von dict benötigen - verwenden groupby + apply + to_dict:

d = df.groupby('key')['id'].apply(list).to_dict() 
print (d) 
{1: ['a1', 'a2', 'a3'], 2: ['a4', 'a5'], 3: ['a6']} 

Oder wenn Bedarf list mit Skalare hinzufügen if/else-apply:

d = df.groupby('key')['id'].apply(lambda x: list(x) if len(x) > 1 else x.iat[0]).to_dict() 
print (d) 
{1: ['a1', 'a2', 'a3'], 2: ['a4', 'a5'], 3: 'a6'} 

Answer 2

Verwenden Sie ein Wörterbuch Verständnis um den groupby Iterator

{n: v.tolist() for n, v in df.groupby('key').id}