Convert Datenrahmen zu einer numpy Matrix

Question

Ich habe Datenrahmen der Form

user_id item_id rating 
1   abc  5 
1   abcd  3 
2   abc  3 
2   fgh  5 

Ich möchte es in eine numpy Matrix

wie

# abc abcd fgh 
[[5, 3, 0] # user_id 1 
[3, 0, 5]] # user_id 2 

umwandeln Kann mir jemand helfen?

Answer 1

können Sie pivot mit fillna, von values zu int und letzte Konvertit Array geworfen verwenden:

arr = df.pivot('user_id', 'item_id', 'rating').fillna(0).astype(int).values 
print (arr) 
[[5 3 0] 
[3 0 5]] 

Eine andere Lösung mit set_index, unstack und values:

arr = df.set_index(['user_id','item_id']).unstack(fill_value=0).values 
print (arr) 
[[5 3 0] 
[3 0 5]]