python 分组合并元素

python 分组合并元素

• 问题描述：将用户点击过的物品进行合并，返回用户点击记录的字典数组.
• 此处用到了python字典进行处理.

1. 实例分析

import pandas as pd

res = {}
data = [[0, 0, 1], [0, 1, 1], [0, 2, 1], [1, 1, 1], [1, 2, 1]] # [u_id, i_id, rating]
for item in data:
    res.setdefault(item[0], []).append(item[1])
print(res)

2. 结果展示

{0: [0, 1, 2], 1: [1, 2]}