python - 将 nD numpy 数组折叠成一维数组

Question

我正在尝试沿特定轴对 nD 数组的值求和，以有效地将其折叠成一维数组。我一直在浏览文档，但未能找到正确的功能。我会尝试用一些代码更好地解释我的问题:

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

我需要的输出是:

array([5,10,15])

我的实际数据是一个 7 MB 的文件，所以我真的不想使用 for 循环。

谢谢你的帮助

Answer 1

只是做

numpy.sum(g, axis=0)

应该可以。