在一个维度上添加numpy 3D数组

Question

我有一个具有以下形状的numpy数组：

(365L, 280L, 300L)

我想在第一维（365）上总结数组，以便得到365个值。

我可以np.sum()，但如何指定哪个轴？

- 编辑：

答案应该有形：（365，）

Answer 1

NumPy版本＆gt; = 1.7

np.sum允许使用import numpy a = numpy.random.random((365L, 280L, 300L)) # just an example s = numpy.sum(a, axis=(1,2)) print s.shape > (365,) 作为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)) 已经是一个numpy数组时才有效。 arr即使您的np.sum是python - arr也可以使用。

NumPy版本＆lt; 1.7：

将元组用作list参数的选项尚未实现，但您可以始终嵌套多个axis或np.sum个调用：

.sum

Answer 2

试试这个：

nullptr