Concat 2 list of dictionaries with same id

Question

I have 2 lists of dictionaries

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}]

Is there a way to concat b to a if the id b is on id a?

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

https://stackoverflow.com/questions/5946236/how-to-merge-multiple-dicts-with-same-key-or-different-key does this answer your question? — , May 22 '22 at 12:39

Sash Sinha · Accepted Answer · 2022-05-22T13:07:26.097

2

You could use the new merging dictionaries feature introduced in 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}]

For older versions of Python you can try use dict unpacking instead:

>>> 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}]

edited May 22 '22 at 13:07

answered May 22 '22 at 12:43

Sash Sinha

18,743
3
23
40

Is there a way to do that in 3.8? That's my current python version – IshikiHIkaru May 22 '22 at 12:52
Or I should upgrade my python version? – IshikiHIkaru May 22 '22 at 12:53
I added a dict unpacking solution that should work for Python 3.8. – Sash Sinha May 22 '22 at 12:58

score 2 · Answer 2 · answered May 22 '22 at 12:52

This should work:

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

It iterates over the the two dict so it is O(n^2), but it is clear and concise. {**item1, **item2} means adds the key value pairs from item1, then the key value pairs from item2. Here, the results will be:

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

score 1 · Answer 3 · answered May 22 '22 at 13:13

There is no direct solution to this problem. But you can use following code:

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 list of dictionaries with same id

3 Answers3