python - 计算 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

我想这会给你预期的输出

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

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