Collapse nD numpy array into a 1D array

Question

I am trying to sum the values of a nD array along a particular axis to effectively collapse it into a 1D array. I have been looking through the docs but haven't been able to find the right function. I will try to explain my question better with some code:

In [46]: g
Out[46]: 
array([[1, 2, 3],
       [1, 2, 3],
       [1, 2, 3],
       [1, 2, 3],
       [1, 2, 3]])

The output I need is:

array([5,10,15])

My actual data is a 7 MB file so I don't really want to use a for loop.

Thank you for your help

tom10 · Accepted Answer · 2012-03-12 20:42:19Z

6

Just doing

numpy.sum(g, axis=0)

should work.

answered Mar 12, 2012 at 20:42

tom10

69.5k11 gold badges133 silver badges143 bronze badges

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

1 Comment

Anake Over a year ago

aaah, I was looking at ma.compress and things.. way off. Thanks!

Collectives™ on Stack Overflow

Collapse nD numpy array into a 1D array

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related