计算numpy数组中每个二维数组的平均值

Question

我有一个如下的 numpy 数组：

b = numpy.array([[[1,2,3], [4,5,6]], [[1,1,1],[3,3,3]]])
print(b)
[[[1 2 3]
  [4 5 6]]

 [[1 1 1]
  [3 3 3]]]

现在我不想计算数组中每个二维数组的平均值。例如

numpy.mean(b[0])
>>> 3.5
numpy.mean(b[1])
>>> 2.0

如何在不使用 for 循环的情况下做到这一点？

Answer 1

np.mean()可以接受轴的参数，所以根据你的使用情况，你可以做以下任一操作

print("Mean of each column:")
print(x.mean(axis=0))
print("Mean of each row:")
print(x.mean(axis=1))

Answer 2

我认为这会给你预期的输出

通过在axis - see doc for more about axis param 中传递 multi-dim

b.mean(axis=(1,2))
array([3.5, 2. ])