如何使用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)

完成