pandas将同一个userID的itemID拼接在一起

**

ItemCF_df = pd.read_csv('./top_3_20221011.csv')[['userID', 'itemID', 'prediction']]
ItemCF_df['item_id'] = ItemCF_df['itemID'].apply(lambda x: [x])
ItemCF_df_new = ItemCF_df.groupby(['userID'])['item_id'].apply(lambda x: np.concatenate(list(x))).reset_index()
ItemCF_df_new['item_id'] = ItemCF_df_new['item_id'].apply(lambda x: json.dumps(x.tolist()))

脚本位置:

/Users/lijinze/PycharmProjects/Jupiter-files/wechat/pandas分组+list格式转换.ipynb