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

应该有用。