sum of all columns in a two dimensional array python

Question

How can I add column values in a two dimenstional array? For example

[[11.9, 12.2, 12.9],
 [15.3, 15.1, 15.1], 
 [16.3, 16.5, 16.5],
 [17.7, 17.5, 18.1]]

I want it to result to:

[61.2,61.3,62.6]

I have tried this so far:

Btotals.append(sum(float(x)) for x in zip(totals))

However it gave me this:

[<generator object <genexpr> at 0x02A42878>]

Is there a reason you can't use numpy for this? With numpy it would be: numpy.sum(a, axis=0). — Akavall
– Akavall, Commented Nov 6, 2015 at 3:57

David Morton · Accepted Answer · 2015-11-06 03:35:07Z

1

You need to unpack the argument to zip first.

a = [[11.9, 12.2, 12.9],
    [15.3, 15.1, 15.1], 
    [16.3, 16.5, 16.5],
    [17.7, 17.5, 18.1]]
result = [sum(x) for x in zip(*a)]
>>> [61.2, 61.3, 62.6]

answered Nov 6, 2015 at 3:35

David Morton

744 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

David Morton Over a year ago

If you want to do anything more complicated, I agree with @Akavall, numpy would be the way to go.

d_unknown Over a year ago

how can I get the sum of square of each element in a?

George · Accepted Answer · 2015-11-06 03:28:33Z

0

If array is a variable containing your 2d array, try

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

answered Nov 6, 2015 at 3:28

George

2,1033 gold badges14 silver badges24 bronze badges

Collectives™ on Stack Overflow

sum of all columns in a two dimensional array python

2 Answers 2

2 Comments

Comments

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

2 Comments

Comments

Related