Create tuple from key-value

Question

I have a list with dictionary inside, with the following keys: id, external_id_client, product_id, as below:

[{'id': 3, 'external_id_client': '7298', 'product_id': 13}, 
{'id': 3, 'external_id_client': '7298', 'product_id': 8}, 
{'id': 4, 'external_id_client': '3', 'product_id': 12}, 
{'id': 4, 'external_id_client': '3', 'product_id': 9}, 
{'id': 5, 'external_id_client': '4', 'product_id': 12}]

I need that if the ID and external_id are repeated and I have information in the product_id key it creates a tuple in the product_id.

Expected output:

[{'id': 3, 'external_id_client': '7298', 'product_id': (13, 8)}, 
{'id': 4, 'external_id_client': '3', 'product_id': (12, 9)}, 
{'id': 5, 'external_id_client': '4', 'product_id': (12)}]

How can I do this?

alex · Accepted Answer · 2022-05-12 13:59:21Z

2

You can use itertools.groupby for this:

from itertools import groupby

lst = [
    {'id': 3, 'external_id_client': '7298', 'product_id': 13}, 
    {'id': 3, 'external_id_client': '7298', 'product_id': 8}, 
    {'id': 4, 'external_id_client': '3', 'product_id': 12}, 
    {'id': 4, 'external_id_client': '3', 'product_id': 9}, 
    {'id': 5, 'external_id_client': '4', 'product_id': 12}
]

result = [
    {
        "id": itemid, 
        "external_id_client": client,
        "product_id": tuple(item["product_id"] for item in group)
    }
    for (itemid, client), group 
        in groupby(lst, key = lambda item: (item["id"], item["external_id_client"]))
]

edited May 12, 2022 at 13:59

alex

11.4k17 gold badges78 silver badges117 bronze badges

answered May 12, 2022 at 13:57

trincot

357k38 gold badges282 silver badges338 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Caroline Leite Over a year ago

Perfect! The only problem is that when it only has a number it has a comma at the end, for example: (9,), is there a way to fix this?

trincot Over a year ago

This is just a notation in the output. A tuple with one member is always encoded like that to make the difference with just a parenthesed expression. It's python syntax.

Collectives™ on Stack Overflow

Create tuple from key-value

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related