如何从数据框架pandas创建列表

Question

我的数据集包含他们购买的usersID和itemsID的列。每个用户可能购买了1件以上的商品。

我需要制作一个列表，以便键将是userID，值将是他购买的itemsID例如，如果user_1购买了[item_20,item_25,item_32]，我的数据集包含该用户的3行，如下所示

row_1= 1,20 row_2= 1,25 row_3= 1,32

我想让我的列表的格式为{1:[20,25,32]}

我想为我的数据集中的所有用户创建一个列表，如上面的例子。

Answer 1

如果我理解正确的话，你想要这样的东西!

下次看看你做了什么会有帮助

df = pd.DataFrame({'user': ['K0', 'K0', 'K2', 'K3', 'K4', 'K5'],
'product': ['A0', 'A1', 'A2', 'A3', 'A4', 'A5']})
my_final_list={}
grouped_df=df.groupby(by=["user"])
for key, item in grouped_df:
products_list=list(grouped_df.get_group(key)["product"])
my_final_list[key]=products_list
print(my_final_list)