Как суммировать значения столбцов с помощью Python - PullRequest
3 голосов
/ 28 марта 2012

У меня есть набор строк, который выглядит следующим образом:

defaultdict(<type 'dict'>, 
{
   u'row1': {u'column1': 33, u'column2': 55, u'column3': 23}, 
   u'row2': {u'column1': 32, u'column2': 32, u'column3': 17}, 
   u'row3': {u'column1': 31, u'column2': 87, u'column3': 18}
})

Я хочу иметь возможность легко получить сумму столбца column1, column2, column3.Было бы здорово, если бы я мог сделать это для любого числа столбцов, получив результат в хэш-карте, которая выглядит как columnName => columnSum.Как вы можете догадаться, я не могу получить суммированные значения из базы данных, поэтому стоит задать вопрос.

Ответы [ 4 ]

7 голосов
/ 28 марта 2012
>>> from collections import defaultdict
>>> x = defaultdict(dict, 
    {
        u'row1': {u'column1': 33, u'column2': 55, u'column3': 23}, 
        u'row2': {u'column1': 32, u'column2': 32, u'column3': 17}, 
        u'row3': {u'column1': 31, u'column2': 87, u'column3': 18}
    }) 

>>> sums = defaultdict(int)
>>> for row in x.itervalues():
        for column, val in row.iteritems():
            sums[column] += val


>>> sums
defaultdict(<type 'int'>, {u'column1': 96, u'column3': 58, u'column2': 174})

О, намного лучше!

>>> from collections import Counter
>>> sums = Counter()
>>> for row in x.values():
        sums.update(row)


>>> sums
Counter({u'column2': 174, u'column1': 96, u'column3': 58}) 
2 голосов
/ 28 марта 2012

Вложенные генераторы + понимание списка делает свое дело:

>>> foo
defaultdict(<type 'dict'>, {u'row1': {u'column1': 33, u'column3': 23, u'column2': 55}, u'row2': {u'column1': 32, u'column3': 17, u'column2': 32}, u'row3': {u'column1': 31, u'column3': 18, u'column2': 87}})
>>> dict(zip(foo.values()[0].keys(), [sum(j[k] for j in (i.values() for _,i in foo.items())) for k in range(3)]))
{u'column1': 96, u'column3': 58, u'column2': 174}
0 голосов
/ 10 августа 2014

Вот еще один ответ, если я могу предложить решение. Сначала поместите ваши данные в матрицу. Затем умножьте матрицу на вектор единиц.

import numpy as np
A = np.random.normal(size = (3,3))

Теперь, чтобы получить сумму столбцов, просто используйте скалярное произведение.

np.dot(A, np.ones(3))

Чтобы сложить строки, а не столбцы, просто транспонируйте матрицу.

np.dot(A.T, np.ones(3))
0 голосов
/ 28 марта 2012

не самый питонический, но здесь:

for val in defaultdict.values() :
   sum1 += val['column1']
   sum2 += val['column2']
   sum3 += val['column3']
final_dict = {'column1' : sum1,'column2' : sum2,'column3' : sum3 }
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...