python - 字典 python 的组列表

Question

如何将字典中相似的键分组到列表中

如果我有

data = [{'quantity': 2, 'type': 'Vip'}, {'quantity': 23, 'type': 'Vip'}, {'quantity': 2, 'type': 'Regular'}, {'quantity': 2, 'type': 'Regular'}, {'quantity': 2, 'type': 'Regular'}, {'quantity': 2, 'type': 'Regular'}]

我希望它像这样输出

res = {'Regular': [{'quantity': 2, 'type': 'Regular'},{'quantity': 2, 'type': 'Regular'},{'quantity': 2, 'type': 'Regular'}], 'Vip': [{'quantity': 23, 'type': 'Vip'},{'quantity': 23, 'type': 'Vip'}]}

这是我试过的代码，但它给了我双倍的 key ，可能是因为循环

 res = defaultdict(list)
 for i in data:
    if len(res) >= 1:
       for q in res:
          if q == i['type']:
            res[q].append(i)
            break
          else:
            res[i['type']].append(i)
            break
  res[i['type']].append(i)

Answer 1

我认为您还没有完全理解defaultdict 的概念。如果在查找时不存在，defaultdict 将生成一个新对象。

所以你可以简单地使用:

from collections import defaultdict

res = defaultdict(list)

for i in data:
    res[i['type']].append(i)

产生:

>>> pprint(res)
defaultdict(<class 'list'>,
            {'Regular': [{'quantity': 2, 'type': 'Regular'},
                         {'quantity': 2, 'type': 'Regular'},
                         {'quantity': 2, 'type': 'Regular'},
                         {'quantity': 2, 'type': 'Regular'}],
             'Vip': [{'quantity': 2, 'type': 'Vip'},
                     {'quantity': 23, 'type': 'Vip'}]})

(pprint 是漂亮打印，但不改变内容)。

请注意，这里我们将字典的引用复制到新列表中，因此我们不会创建新字典。此外，结果是一个 defaultdict。我们可以使用 dict(res) 将其转换为 vanilla 字典。