将 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)

应该可以。