我有一個像這樣的熊貓數據框：TransactionID ProductID1 1321 2561 9852 3213 4513 219我想按“TransactionID”分組并將“ProductID”分配給一個列表，如下所示：list = [[132, 256, 985], [321], [451, 291]]執行此任務的正確方法是什么？
2 回答
ABOUTYOU
TA貢獻1812條經驗 獲得超5個贊
以下不是很好，但可以工作。
result = [list(i.ProductID) for i in dict(list(df.groupby("TransactionID"))).values()]
米琪卡哇伊
TA貢獻1998條經驗 獲得超6個贊
像這樣的事情可能會有所幫助。
您只需按 將它們分組TransactionID，然后從中取出ProductID并轉換為列表
grouped_list = list(df.groupby('TransactionID')['ProductID'].apply(list))正如評論中提到的，使用“list”作為變量名并不好。這是因為您將重新定義 list 命令的原始功能，將其設置為您從 df 中提取的分組列表。
添加回答
舉報
0/150
提交
取消