numpy的mean函数用于求平均值。

全部元素的平均值

import numpy as np

a = np.array([[1, 2], [3, 4]])
print(np.mean(a))

执行结果：

2.5

对各列求均值

import numpy as np

a = np.array([[1, 2], [3, 4]])
print(np.mean(a, axis=0))

执行结果：

[2. 3.]

对各行求均值

import numpy as np

a = np.array([[1, 2], [3, 4]])
print(np.mean(a, axis=1))

执行结果：

[1.5 3.5]