如何通过对值求和而不使用 for 循环将 2d numpy 数组转换为 1d numpy 数组？

Question

是否有一个 numpy 函数可以将 2d numpy 数组组合成 1d numpy 数组。我想在不使用 for 循环的情况下做到这一点。

例子：

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