Concat 2具有相同ID的字典列表

Question

我有 2 个词典列表

a = [{'id':1, 'name':'John Doe'}, {'id':2, 'name':'Jane Doe'}, {'id':4, 'name':'Sample Doe'}]
b = [{'id':1, 'rating':9}, {'id':2, 'rating':7}, {'id':3, 'rating':8}]

如果 id b 在 id a 上，有没有办法将 b 连接到 a？

[{'id':1, 'name':'John Doe', 'rating':9}, {'id':2, 'name':'Jane Doe', 'rating':7}, {'id':4, 'name':'Sample Doe', 'rating':0}]

Answer 1

您可以使用 Python 3.9 中引入的新合并词典功能：

>>> a = [{'id': 1, 'name': 'John Doe'}, {'id': 2, 'name': 'Jane Doe'}, {'id': 4, 'name': 'Sample Doe'}]
>>> b = [{'id': 1, 'rating': 9}, {'id': 2, 'rating': 7}, {'id': 3, 'rating': 8}]
>>> b_id_to_d = {d['id']: d for d in b}  # Create for O(1) lookup time by id.
>>> b_id_to_d
{1: {'id': 1, 'rating': 9}, 2: {'id': 2, 'rating': 7}, 3: {'id': 3, 'rating': 8}}
>>> c = [d | b_id_to_d.get(d['id'], {'rating': 0}) for d in a]
>>> c
[{'id': 1, 'name': 'John Doe', 'rating': 9}, {'id': 2, 'name': 'Jane Doe', 'rating': 7}, {'id': 4, 'name': 'Sample Doe', 'rating': 0}]

对于 Python 的旧版本，您可以尝试使用 dict 解包：

>>> c = [{**d, **b_id_to_d.get(d['id'], {'rating': 0})} for d in a]
>>> c
[{'id': 1, 'name': 'John Doe', 'rating': 9}, {'id': 2, 'name': 'Jane Doe', 'rating': 7}, {'id': 4, 'name': 'Sample Doe', 'rating': 0}]

Answer 2

这应该有效：

[{**item1, **item2} for item1 in a for item2 in b if item1['id'] == item2['id']]

它遍历了两个字典，所以它是 O(n^2)，但它清晰简洁。 {**item1, **item2} 表示添加 item1 的键值对，然后添加 item2 的键值对。在这里，结果将是：

[{'id': 1, 'name': 'John Doe', 'rating': 9},
 {'id': 2, 'name': 'Jane Doe', 'rating': 7}]

Answer 3

这个问题没有直接的解决方案。但是你可以使用下面的代码：

a = [{'id':1, 'name':'John Doe'}, {'id':2, 'name':'Jane Doe'}]
b = [{'id':1, 'rating':9}, {'id':2, 'rating':7}, {'id':3, 'rating':8}]
key_pos_mapping = {}
for index,dict in enumerate(a):
    key_pos_mapping[dict['id']] = index
 
for dict in b:
    if( dict['id'] in key_pos_mapping.keys()):
        dict.update(a[key_pos_mapping[dict['id']]])
    else:
        b.remove(dict)

Concat 2具有相同ID的字典列表

Concat 2 list of dictionaries with same id

python

list