Как преобразовать массив 2d numpy в массив 1d numpy, суммируя значения и не используя для l oop?

Question

Существует ли функция numpy, которая может объединять массив 2d numpy в массив 1d numpy. Я хочу сделать это без использования для l oop.

Пример:

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

Answer 1

Просто используйте метод ndarray sum вдоль оси строки:

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

arr1d = arr2d.sum(axis=0)

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