np.mean()

Среднее арифметическое элементов (всего массива или по оси).

Сигнатураnp.mean(a, axis=None)

Считает среднее. С axis — по столбцам/строкам.

import numpy as np
a = np.array([[1, 2], [3, 4]])
print(np.mean(a))          # 2.5
print(np.mean(a, axis=1))  # [1.5 3.5]