Question

我有字典列表，每个字典都有属性person_id

[{'person_id': 1, 'something..': 'else'}, {'person_id': 3, 'something..': 'else'}, {'person_id': 1, 'something..': 'else'}]

如何将具有相同“person_id”的所有词典分组到列表中，并将以前的词典列表拆分为词典列表列表（内部列表仅包含具有相同“person_id”的词典）。

[[{'person_id': 1, 'something..': 'else'}, {'person_id': 1, 'something..': 'else'}], [{'person_id': 3, 'something..': 'else'}]]

我是通过简单的迭代完成的，并且检查已经列出了那些字典，但在Python中是否有有效的方法？

Answer 1

您的输入未按person_id排序，因此请使用collections.defaultdict()方法循环播放：

from collections import defaultdict

grouped = defaultdict(list)

for person in inputlist:
    grouped[person['person_id']].append(person)

grouped = grouped.values()

这会将输入列表排序到存储桶中，然后将所有存储桶放回到外部列表中。如果您需要按id排序输出，可以使用：

grouped = [grouped['pid'] for pid in sorted(grouped)]

如果您需要按排序顺序排列原始列表，或者按排序顺序生成，您还可以使用itertools.groupby()进行分组：

from itertools import groupby
from operator import itemgetter

grouped = [list(g) for p, g in groupby(inputlist, key=itemgetter('person_id'))]

但这确实需要先排序inputlist。

将字典列表分组到具有相同属性值的字典列表

1 个答案: