NumPy 函数

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]