np.mean参数和用法

np.mean() 是 Numpy 库中的函数,用于计算数组中元素的平均值。

语法: np.mean(a, axis=None, dtype=None, out=None, keepdims=)

a:要计算平均值的数组。
axis:计算平均值的轴,若不指定,则计算整个数组的平均值。
dtype:计算平均值的数据类型。
out:用于放置结果的可选数组。
keepdims:指定是否保持轴数不变。

import numpy as np

a = np.array([[1, 2], [3, 4]])
print(np.mean(a))  # 打印出 2.5
print(np.mean(a, axis=0))  # 打印出 [2. 3.]
print(np.mean(a, axis=1))  # 打印出 [1.5 3.5]