如何使用numpy.mean在特定维度上求平均值？

Question

我有一个名为 POS 的矩阵，其形式为 (10,132)，我需要对前 10 个元素进行平均，使我的平均矩阵具有 (1,132) 的形式

我已经尝试过做

means = pos.mean (axis = 1)

或者

menas = np.mean(pos)

但是第一种情况的结果是 (10,) 的矩阵，而第二种情况是一个简单的数字

我期望输出形状为 (1,132) 的矩阵

Answer 1

解决方案是指定正确的轴并使用

keepdims=True

，一些评论者指出了这一点。

这可以通过

pos.mean(axis = 0,keepdims=True)

或

np.mean(pos,axis=0,keepdims=True)

来完成