在一維上添加numpy 3D數組

Question

我有一個具有以下形狀的numpy數組：

(365L, 280L, 300L)

我想對第一個維度（365）上的數組求和，以便得到365個值。

我可以做np.sum() ，但是如何指定哪個軸呢？

- 編輯：

答案應為：（365，）

Answer 1

NumPy版本> = 1.7

np.sum允許使用tuple of integer作為axis參數來一次計算沿多個軸的和：

import numpy as np
arr = ... # somearray

np.sum(arr, axis=(1, 2)) # along axis 1 and 2 (remember the first axis has index 0)
np.sum(arr, axis=(2, 1)) # the order doesn't matter

或直接使用數組的sum方法：

arr.sum(axis=(1, 2))

僅當arr已經是一個numpy數組時，后者才有效。 np.sum工程，即使你的arr是python- list 。

NumPy版本<1.7：

使用元組作為axis參數的選項尚未實現，但您始終可以嵌套多個np.sum或.sum調用：

np.sum(np.sum(arr, axis=1), axis=1)  # Nested sums
arr.sum(axis=2).sum(axis=1)          # identical but more "sequential" than "nested"

Answer 2

嘗試這個：

import numpy

a = numpy.random.random((365L, 280L, 300L)) # just an example

s = numpy.sum(a, axis=(1,2))

print s.shape

> (365,)