How to convert a 2d numpy array into a 1d numpy array by summing the values and not using for loop?

Question

Is there a numpy function which can combine a 2d numpy array into a 1d numpy array. I want to do it without using a for loop.

Example:

[[1 0 0 0 0], [0 1 0 0 0]] => [1 1 0 0 0]

yacola · Accepted Answer · 2020-04-14 19:17:01Z

1

Just use the ndarray method sum along row axis:

arr2d = np.array([[1, 3, 8, 2, 0], [0, 1, 0, 5, 1]])

arr1d = arr2d.sum(axis=0)

>>> array([1, 4, 8, 7, 1])

answered Apr 14, 2020 at 19:17

yacola

3,0332 gold badges13 silver badges30 bronze badges

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

1 Answer 1