获取 numpy ndarray 的平均值

Question

我有一个A = (8, 64, 64, 64, 1)numpy.ndarray 的形状。我们可以使用np.meansornp.average来计算一个 numpy 数组的均值。但我想获得 8 个(64,64,64)数组的方法。也就是说，我只想要 8 个值，从(64,64,64). 当然我可以使用 for 循环，或者使用[np.means(A[i]) for i in range(A.shape[0])]. 我想知道是否有任何 numpy 方法可以做到这一点

Answer 1

您可以使用np.means 轴 kwarg：

np.mean(A, (1, 2, 3, 4))

也同样适用np.average。