2017-06-01 110 views
1

對熊貓比較新 - 我有一個包含電影ID,用戶ID,等級和日期的數據框。我已經按用戶ID和日期排序,並具有下面的數據框。將轉置分組的熊貓數據框輸出爲CSV

https://i.stack.imgur.com/fqSZ6.png

我所需的輸出是指每個用戶一行,與該用戶已評級排序按時間順序從左到右的所有電影CSV。例如:

452 4 33 6581 
56 
121 69 98 802 555 
. 
. 

master_sample.sort_values(['User ID','Date']).groupby('User ID') 

然而,通過用戶ID分組後,我收到了GROUPBY對象,我不能確定如何遍歷並輸出到CSV。我已經嘗試了使用get_group在原始df/iterated分組的df上的pivot。

任何指針,將不勝感激!

回答

1

試試這個:

master_sample.sort_values('Date') \ 
      .groupby('User ID', as_index=False)['Movie ID'] \ 
      .apply(' '.join) 
+0

我很喜歡它! – piRSquared

+0

@piRSquared,謝謝:) – MaxU

+0

謝謝 - 完美的作品! – Matt