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

应该管用.