DjangoBB LoFi version

Полная версия: Как сложить значения словарей в списке

Начало » Python для новичков » Как сложить значения словарей в списке

slepov

Апрель 16, 2015 17:10:16

Есть список содержащий словари:

[{'id': 1, 'val':1}, {'id': 2, 'val':3}, {'id': 1, 'val':4} ]

Если значения `id` одинаковы, то необходимо сложить значения `val`, то есть получить

[{'id': 1, 'val':5}, {'id': 2, 'val':3} ]

botinag

Апрель 16, 2015 20:20:23

lst = [{'id': 1, 'val':1}, {'id': 2, 'val':3}, {'id': 1, 'val':4}]
def f(lst):
    dct = {}
    for x in lst:
        if x['id'] in dct:
            dct[x['id']] += x['val']
        else:
            dct[x['id']] = x['val']
    return [{'id': x, 'val': y} for x, y in dct.iteritems()]

>>> f(lst)
[{'id': 1, 'val': 5}, {'id': 2, 'val': 2}]
>>> lst = [{'id': 1, 'val':1}, {'id': 2, 'val':3}, {'id': 1, 'val':4}]
>>> f(lst)
[{'id': 1, 'val': 5}, {'id': 2, 'val': 3}]

terabayt

Апрель 17, 2015 00:55:09

botinag

if x['id'] in dct:
    dct[x['id']] += x['val']
else:
    dct[x['id']] = x['val']

dct[x['id']] = dct.get(x['id'], 0) + x['val']

slepov

Апрель 17, 2015 08:49:59

А если задачу слегка усложнить, например словари имеют несколько большее количество пар ключ:значение.

[{'id': 1, 'val':1, 'a': 'd'}, {'id': 2, 'val':3, 'a': 'b' }, {'id': 1, 'val':4, 'a': 'c'} ]

terabayt

Апрель 17, 2015 09:07:24

slepov
А если задачу слегка усложнить, например словари имеют несколько большее количество пар ключ:значение.

и что длать с новыми парами? проверять или складывать?

slepov

Апрель 17, 2015 09:18:16

terabayt
и что длать с новыми парами? проверять или складывать?

Упс… Ошибся в примере.

[{'id': 1, 'val':1, 'a': 'd'}, {'id': 2, 'val':3, 'a': 'b' }, {'id': 1, 'val':4, 'a': 'd'} ]

Ничего не делать, просто возвращать . Причем пар может быть произвольное количество, но всегда одинаковое во всех словарях списка. При этом будем считать что все пары кроме val имеют зависимость от пары `id`, то есть для одинаковых `id` cчитаем, что все остальные пары в словаре одинаковы.

bismigalis

Апрель 17, 2015 10:58:51

data = [{'id': 1, 'val':1, 'a': 'd'}, {'id': 2, 'val':3, 'a': 'b' }, {'id': 1, 'val':4, 'a': 'c'} ]
def f(d, x):
    k = x['id']
    if k in d:
        d[k]['val'] += x['val']
    else:
        d[k] = x
    return d
print reduce(f, data, {}).values()

slepov

Апрель 17, 2015 11:11:42

Спасибо это ОНО!

killik_23

Апрель 19, 2017 17:44:34

Подскажите, как сделать тоже самое, но только не складывать значение ‘val’, а выбрать наибольшее из списка словарей?

killik_23

Апрель 19, 2017 17:59:27

Решил так, скажите, это оптимальный вариант?

 data = [{'id': 1, 'val':1, 'a': 'd'}, {'id': 2, 'val':3, 'a': 'b' }, {'id': 1, 'val':4, 'a': 'c'} ]
def f(d, x):
    k = x['id']
    if k in d:
         d[k]['val'] = max(x['val'] for d in data)
    else:
        d[k] = x
    return d
print reduce(f, data, {}).values()