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]