如何从 dataframe pandas 创建列表

Question

我的数据集包含他们购买的用户 ID 和项目 ID 列。 每个用户可能购买了超过 1 件商品。

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

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

我希望我的列表包含 fromat {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)