轉換數據幀到numpy的矩陣

Question

我有形式

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

我想將其轉換成作爲

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

誰能幫助numpy的矩陣這樣的數據幀？

Answer 1

您可以通過values使用pivot與fillna，轉換爲int和最後轉換到數組：

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

與set_index，unstack和values另一種解決方案：

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