NumPy mean() 函数用于计算沿指定轴的算术平均值。默认情况下,在展平数组上计算平均值,否则在指定轴上计算平均值。
语法
numpy.mean(a, axis=None, dtype=None, out=None, keepdims=<no value>)
参数
a | 必填。 指定一个包含所需平均值的数字的数组。如果 a 不是数组,则尝试转换。 |
axis | 可选。 指定计算平均值所沿的一个或多个轴。默认值是计算展平数组的平均值。 |
dtype | 可选。 指定计算平均值的数据类型。对于整数输入,默认值为 float64。对于浮点输入,它与输入数据类型相同。 |
out | 可选。 指定结果的输出数组。默认值为"无"。如果提供,它必须具有与输出相同的形状。 |
keepdims | 可选。 如果将此设置为 True,则缩小的轴将作为大小为 1 的维度保留在结果中。使用此选项,结果将针对输入数组正确广播。使用默认值时,keepdims 不会传递到 ndarray 子类的 Mean 方法,但任何非默认值都会传递到。如果子类方法没有实现keepdims,则会引发异常。 |
返回值
当 out=None 时,返回包含平均值的数组,否则返回对输出数组的引用。
示例:所有值的平均值
在下面的示例中,mean()函数用于计算数组中存在的所有值的平均值。
import numpy as np
Arr = np.array([[10,20],[30, 40]])
print("Array is:")
print(Arr)
#所有值的平均值
print("\n所有值的平均值:", np.mean(Arr))
上述代码的输出将是:
Array is:
[[10 20]
[30 40]]
所有值的平均值: 25.0
示例:带有轴的mean()参数
当提供轴参数时,将计算指定轴上的平均值。考虑以下示例。
import numpy as np
Arr = np.array([[10,20,30],[70,80,90]])
print("Array is:")
print(Arr)
#沿轴平均值=0
print("\n沿轴平均值=0")
print(np.mean(Arr, axis=0))
#沿轴平均值=1
print("\n沿轴平均值=1")
print(np.mean(Arr, axis=1))
上述代码的输出将是:
Array is:
[[10 20 30]
[70 80 90]]
沿轴平均值=0
[40. 50. 60.]
沿轴平均值=1
[20. 80.]
示例:带有 dtype 参数的mean()
可以提供 dtype 参数来获取所需数据类型的结果。
import numpy as np
Arr = np.array([[10,20],[80,90]])
print("Array is:")
print(Arr)
#mean 沿 axis=0 且 dtype=complex
print("\n沿轴平均值=0 (dtype=complex)")
print(np.mean(Arr, axis=0, dtype=complex))
上述代码的输出将是:
Array is:
[[10 20]
[80 90]]
沿轴平均值=0 (dtype=complex)
[45.+0.j 55.+0.j]