Jak mogę podsumować kolumnę listy?

Question

Mam tablicę Pythona, tak jak:

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

mogę dodać wiersz wykonując sum(array[i]), jak mogę podsumować kolumnę za pomocą podwójnej pętli for?

I.E. dla pierwszej kolumny, mogę dostać 2, potem 4, potem 6.

Answer 1

Korzystanie z for pętlę (w generator expression):

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

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

Answer 2

Nie trzeba pętlę, należy zip() transpozycji listę, a następnie podjąć żądaną kolumnę:

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

(Uwaga w 2.x, zip() zwraca listę, więc nie potrzebujesz połączenia list()).

Edit: Najprostszym rozwiązaniem tego problemu, bez użycia zip(), będzie prawdopodobnie:

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

Właśnie pętli wierszy, biorąc elementu i dodanie go do naszego sumie.

Jest to jednak mniej wydajne i raczej bezsensowne, ponieważ mamy wbudowane funkcje, które umożliwiają nam to. Zasadniczo użyj zip().

Answer 3

można użyć zip():

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

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

lub prosty dla pętli:

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]))] 

To powinno wystarczyć. len(array[0]) to liczba kolumn, więc i iteruje za ich pośrednictwem. Wyrażenie generatora row[i] for row in array przechodzi przez wszystkie wiersze i wybiera pojedynczą kolumnę dla każdego numeru kolumny.

Answer 5

Spróbuj tego:

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

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

zip function description

Answer 6

Możesz być zainteresowany numpy, który ma więcej zaawansowanych funkcji tablicowych. Jednym z nich jest łatwo podsumować kolumnę:

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

Myślę, że najprostszym sposobem jest taka:

sumcolumn=data.sum(axis=0) 

print (sumcolumn) 

Answer 8

Można użyć numpy:

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