Как я могу суммировать столбец списка?

Question

У меня есть массив Python, например, так:

[[1,2,3], 
[1,2,3]] 

Я могу добавить строку, выполнив sum(array[i]), как я могу суммировать столбец, используя двойной цикл?

I.E. для первого столбца, я мог бы получить 2, затем 4, затем 6.

Answer 1

Использование for петли (в generator expression):

data = [[1,2,3], 
     [1,2,3]] 

column = 1 
print sum(row[column] for row in data) 

Answer 2

Вам не нужен цикл, используйте zip() транспонировать список, а затем взять нужный столбец:

sum(list(zip(*data)[i])) 

(Примечание в 2.x, zip() возвращает список, так что вам не нужен звонок list()).

Edit: Самое простое решение этой проблемы, без использования zip(), вероятно, будет:

column_sum = 0 
for row in data: 
    column_sum += row[i] 

Мы просто цикл по строкам, принимая элемент и добавить его в нашем общем.

Это, однако, менее эффективно и довольно бессмысленно, поскольку у нас есть встроенные функции для этого. В общем, используйте zip().

Answer 3

вы можете использовать zip():

In [16]: lis=[[1,2,3], 
    ....: [1,2,3]] 

In [17]: map(sum,zip(*lis)) 
Out[17]: [2, 4, 6] 

или просто для петель:

In [25]: for i in xrange(len(lis[0])): 
    summ=0 
    for x in lis: 
     summ+=x[i] 
    print summ 
    ....:  
2 
4 
6 

Answer 4

[sum(row[i] for row in array) for i in range(len(array[0]))] 

Это должно сделать это. len(array[0]) - количество столбцов, поэтому i выполняет итерации. Выражение генератора row[i] for row in array проходит через все строки и выбирает один столбец для каждого номера столбца.

Answer 5

Попробуйте это:

a = [[1,2,3], 
    [1,2,3]] 

print [sum(x) for x in zip(*a)] 

zip function description

Answer 6

Вы можете быть заинтересованы в numpy, которая имеет более расширенные функции массива. Одним из них является легко суммировать столбец:

from numpy import array 

a = array([[1,2,3], 
[1,2,3]]) 

column_idx = 1 
a[:, column_idx].sum() # ":" here refers to the whole array, no filtering. 

Answer 7

Я думаю, самый простой способ заключается в следующем:

sumcolumn=data.sum(axis=0) 

print (sumcolumn) 

Answer 8

Вы можете использовать NumPy:

import numpy as np 
a = np.array([[1,2,3],[1,2,3]]) 
a.sum(0)