Группировать и объединять список словарей по нескольким ключам

Вот один из способов сделать это:

Я сначала построил dict, используя кортеж в качестве составного ключа, а затем составил новый список из этого dict. Вы можете написать понимание для дальнейшего сокращения строк и оптимизации, надеюсь, это поможет.

new_dict = {}

for item in lst:
    try: # if already exists then append to the list
        new_dict.get((item['number'], item['favorite']))
        new_dict[(item['number'], item['favorite'])].append(item['color'])
    except KeyError: # if it doesn't then create a new entry to that key
        new_dict[(item['number'], item['favorite'])] = [item['color']]


final_list = []
for k, v in new_dict.items(): # keep appending dicts to our list
    final_list.append({'number': k[0], 'favorite': k[1], 'color':set(v)})

print(final_list)

Выходы:

[{'number': 1, 'favorite': False, 'color': {'green', 'red'}}, {'number': 1, 'favorite': True, 'color': {'red'}}, {'number': 2, 'favorite': False, 'color': {'red'}}]

Мой друг сделал следующую функцию, чтобы решить эту проблему, без использования каких-либо внешних библиотек:

def uniqifyColors(l):
    for elem in l:
        for item in l:
            if elem['number'] == item['number'] and elem['favorite'] == item['favorite']:
                for clr in item['color']:
                    if clr not in elem['color']:
                        elem['color'].append(clr)
    return l

После использования этой функции Python он просто выполнил тривиальное uniqify, чтобы получить уникальные результаты из список. Однако он не сохраняет один цвет в виде строки, а представляет собой список с одним элементом.

Решением в чистом Python было бы использование defaultdict с составным ключом. Вы можете использовать это, чтобы объединить ваши ценности. После этого вы можете снова создать список из этого словаря.

from collections import defaultdict

dct = defaultdict([])

for entry in lst:
    dct[(entry['number'], entry['favorite'])].append(entry['color'])

lst = [{'number': key[0], 'favorite': key[1], color: value if len(value) > 1 else value[0]}
    for key, value in dct.items()]

Вы можете использовать упорядоченный словарь со значениями по умолчанию set. ^{1 Затем итерируйте свой список словарей, используя (number, favorite) в качестве ключей. Это работает, так как кортежи являются хэшируемыми и поэтому могут использоваться в качестве ключей словаря.}

Хорошей практикой является использование последовательной структуры. Таким образом, вместо того, чтобы иметь строки для отдельных значений и наборы для нескольких, используйте наборы повсюду:

from collections import OrderedDict, defaultdict

class DefaultOrderedDict(OrderedDict):
    def __missing__(self, k):
        self[k] = set()
        return self[k]

d = DefaultOrderedDict()  # Python 3.7+: d = defaultdict(set)

for i in lst:
    d[(i['number'], i['favorite'])].add(i['color'])

res = [{'number': num, 'favorite': fav, 'color': col} for (num, fav), col in d.items()]

print(res)
# [{'color': {'green', 'red'}, 'favorite': False, 'number': 1},
#  {'color': {'red'}, 'favorite': True, 'number': 1},
#  {'color': {'red'}, 'favorite': False, 'number': 2}]

Если вы настаиваете на наличии разных типов в зависимости от количества цветов, вы можете переопределить понимание списка, чтобы использовать троичный оператор:

res = [{'number': num, 'favorite': fav, 'color': next(iter(col)) if len(col) == 1 else col} \
       for (num, fav), col in d.items()]

print(res)
# [{'color': {'green', 'red'}, 'favorite': False, 'number': 1},
#  {'color': 'red', 'favorite': True, 'number': 1},
#  {'color': 'red', 'favorite': False, 'number': 2}]

^{1 Этот момент заслуживает внимания в версиях Python до 3.7, где словари не гарантированно упорядочены по вставке. В Python 3.7+ вы можете воспользоваться порядком вставки и просто использовать dict или подкласс dict, например collections.defaultdict.}

Используя чистый python, вы можете сделать вставку в OrderedDict, чтобы сохранить порядок вставки:

from collections import OrderedDict

d = OrderedDict()
for l in lst:
    d.setdefault((l['number'], l['favorite']), set()).add(l['color'])

[{'number': k[0], 'favorite': k[1], 'color': v.pop() if len(v) == 1 else v} 
    for k, v in d.items()]   
# [{'color': {'green', 'red'}, 'favorite': False, 'number': 1},
#  {'color': 'red', 'favorite': True, 'number': 1},
#  {'color': 'red', 'favorite': False, 'number': 2}]

Это также можно сделать довольно просто, используя API панд GroupBy: [116 ]

import pandas as pd

d = (pd.DataFrame(lst)
       .groupby(['number', 'favorite'])
       .color
       .agg(set)
       .reset_index()
       .to_dict('r'))
d
# [{'color': {'green', 'red'}, 'favorite': False, 'number': 1},
#  {'color': {'red'}, 'favorite': True, 'number': 1},
#  {'color': {'red'}, 'favorite': False, 'number': 2}]

Если требуется условие строки для одного элемента, вы можете использовать

[{'color': (lambda v: v.pop() if len(v) == 1 else v)(d_.pop('color')), **d_} 
     for d_ in d]
# [{'color': {'green', 'red'}, 'favorite': False, 'number': 1},
#  {'color': 'red', 'favorite': True, 'number': 1},
#  {'color': 'red', 'favorite': False, 'number': 2}]