как усреднить в определенном измерении с 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)