2017-05-30 33 views
5

Amacım basit ancak mümkün olup olmadığından emin değil. Tekrar üretilebilir örnek:Pandalar: Sayma ve sayma değerlerine sahip tablo oluşturmak için grupby

Eğer bundan gidebilir miyim:

raw_data = {'score': [1, 3, 4, 4, 1, 2, 2, 4, 4, 2], 
     'player': ['Miller', 'Jacobson', 'Ali', 'George', 'Cooze', 'Wilkinson', 'Lewis', 'Lewis', 'Lewis', 'Jacobson']} 
df = pd.DataFrame(raw_data, columns = ['score', 'player']) 
df 

    score player 
0 1  Miller 
1 3  Jacobson 
2 4  Ali 
3 4  George 
4 1  Cooze 
5 2  Wilkinson 
6 2  Lewis 
7 4  Lewis 
8 4  Lewis 
9 2  Jacobson 

bu için: a groupby Via

 score col_1  col_2  col_3  col_4  
score 
1  2  Miller  Cooze  n/a   n/a 
2  3  Wilkinson Lewis  Jacobson n/a 
3  1  Jacobson n/a   n/a   n/a 
4  4  Ali   George  Lewis  Lewis 

?

Bu kadar df.groupby(['score']).agg({'score': np.size}) alabilirim, ancak sütun değerleriyle yeni sütunların nasıl oluşturulacağını çözemiyorum.

cevap

6

Ben

Seçenek 1

g = df.groupby('score').player 
g.size().to_frame('score').join(g.apply(list).apply(pd.Series).add_prefix('col_')) 

     score  col_0 col_1  col_2 col_3 
score           
1   2  Miller Cooze  NaN NaN 
2   3 Wilkinson Lewis Jacobson NaN 
3   1 Jacobson  NaN  NaN NaN 
4   4  Ali George  Lewis Lewis 

Seçenek 2 ben `pop` kullanılan @MaxU

d1 = df.groupby('score').agg({'score': 'size', 'player': lambda x: tuple(x)}) 
d1.join(pd.DataFrame(d1.pop('player').values.tolist()).add_prefix('col_')) 

     score  col_0 col_1  col_2 col_3 
score           
1   2  Miller Cooze  NaN NaN 
2   3 Wilkinson Lewis Jacobson NaN 
3   1 Jacobson  NaN  NaN NaN 
4   4  Ali George  Lewis Lewis 
+2

ile çıkış çoğaltabilirsiniz! Kendimle çok gurur duyuyorum :-) – piRSquared

+0

Her ikisi de muhteşem. Seçenek 2 için doğru çıktıyı alamıyorum (muhtemelen Python 3 kullanıyorum çünkü). Yine de, seçenek 1 mükemmeldir. – RDJ