PythonのDictの同じIDを持つアイテムの統合

Question

前提・実現したいこと

Pythonで辞書の中の同じIDを持つアイテムを統合する方法について教えて下さい。
具体的な仕様としては、指定したキーの値が同じ（複数の／単独の）アイテムを一つのアイテムに統合するというもので、キー毎に値が同じものは追加せず、値が違うものは配列として両方を記録するというものを想定しています。

入力データ

{'ID': 'A1', 'name': 'abc', 'url': 'https://example.com/123'},
{'ID': 'A2', 'name': 'def', 'url': 'https://example.com/456'},
{'ID': 'A3', 'name': 'ghi', 'url': 'https://example.com/789'},
{'ID': 'A1', 'name': 'ab and c', 'url': 'https://example.com/123'},
{'ID': 'A2', 'name': 'def', 'url': 'https://other-example.com/456'},

変換後のデータ

{'ID': 'A1', 'name':['abc', 'ab and c'], 'url': 'https://example.com/123'},
{'ID': 'A2', 'name': 'def', 'url': ['https://example.com/123', 'https://other-example.com/456']},
{'ID': 'A3', 'name': 'ghi', 'url': 'https://example.com/789'},

試したこと

itertoolsのgroupbyなどで値の更新（上書き）などはできたのですが、配列化がうまくいきませんでした。
ご教示のほど、よろしくお願い致します。

Answer 1

普通にやるとこうなります。
他にあまり良い方法も思いつきません。

入力データ

dicts = [{'ID': 'A1', 'name': 'abc', 'url': 'https://example.com/123'},
{'ID': 'A2', 'name': 'def', 'url': 'https://example.com/456'},
{'ID': 'A3', 'name': 'ghi', 'url': 'https://example.com/789'},
{'ID': 'A1', 'name': 'ab and c', 'url': 'https://example.com/123'},
{'ID': 'A2', 'name': 'def', 'url': 'https://other-example.com/456'}]

変換コード

def collect(dlist, key):
    values = list({d[key] for d in dlist if key in d})
    if len(values) == 1:
        return values[0]
    else:
        return values

def unite(dlist, keys):
    return {key : collect(dlist, key) for key in keys}

def make_dict_list(dicts, IDs, keys):
    return [unite([d for d in dicts if d['ID'] == ID], keys) for ID in IDs]

keys = set(sum([list(d.keys()) for d in dicts], []))
IDs = {d['ID'] for d in dicts}

results = make_dict_list(dicts, IDs, keys)

実行結果

>>> results = make_dict_list(dicts, IDs, keys)
>>>
>>> for r in results:
...     print(r)
...
{'name': ['ab and c', 'abc'], 'url': 'https://example.com/123', 'ID': 'A1'}
{'name': 'def', 'url': ['https://example.com/456', 'https://other-example.com/456'], 'ID': 'A2'}
{'name': 'ghi', 'url': 'https://example.com/789', 'ID': 'A3'}

Comments

[etsuji112]

ありがとうございました、とても勉強になりました！