numpy.mean#

numpy.mean(a, axis=None, dtype=None, out=None, keepdims=<no value>, *, where=<no value>)[source]#

沿指定轴计算算术平均值。

返回数组元素的平均值。默认情况下,平均值是针对展平的数组计算的,否则是沿指定轴计算的。对于整数输入,使用 float64 中间值和返回值。

参数:
aarray_like

包含需要计算平均值的数字的数组。如果 a 不是数组,则会尝试进行转换。

axisNone 或 int 或 int 元组,可选

计算平均值的轴。默认为计算展平数组的平均值。

如果这是一个整数元组,则平均值将跨多个轴执行,而不是像以前那样只跨一个轴或所有轴。

dtype数据类型,可选

用于计算平均值的数据类型。对于整数输入,默认值为 float64;对于浮点输入,与输入 dtype 相同。

outndarray,可选

用于存放结果的备用输出数组。默认值为 None;如果提供,其形状必须与预期输出相同,但如果需要,类型会进行转换。有关更多详细信息,请参阅 输出类型确定。有关更多详细信息,请参阅 输出类型确定

keepdimsbool,可选

如果设置为 True,则被缩减的轴将作为大小为一的维度保留在结果中。使用此选项,结果将与输入数组正确广播。

如果传递了默认值,则 keepdims 将不会传递给 ndarray 子类的 mean 方法,但任何非默认值都会传递。如果子类的方法未实现 keepdims,则会引发任何异常。

wherebool 的 array_like,可选

要包含在平均值中的元素。有关详细信息,请参阅 reduce

1.20.0 版新增。

返回:
mndarray,参见上面的 dtype 参数

如果 out=None,则返回包含平均值的新数组;否则,返回对输出数组的引用。

另请参阅

average

加权平均值

std, var, nanmean, nanstd, nanvar

备注

算术平均值是沿轴的元素之和除以元素数量。

请注意,对于浮点输入,平均值使用与输入相同的精度进行计算。根据输入数据,这可能导致结果不准确,特别是对于 float32(请参阅下面的示例)。使用 dtype 关键字指定更高精度的累加器可以缓解此问题。

默认情况下,float16 结果使用 float32 中间值计算,以获得额外精度。

示例

>>> import numpy as np
>>> a = np.array([[1, 2], [3, 4]])
>>> np.mean(a)
2.5
>>> np.mean(a, axis=0)
array([2., 3.])
>>> np.mean(a, axis=1)
array([1.5, 3.5])

在单精度下,mean 可能不准确

>>> a = np.zeros((2, 512*512), dtype=np.float32)
>>> a[0, :] = 1.0
>>> a[1, :] = 0.1
>>> np.mean(a)
np.float32(0.54999924)

在 float64 中计算平均值更准确

>>> np.mean(a, dtype=np.float64)
0.55000000074505806 # may vary

可以计算 timedelta64 的平均值

>>> b = np.array([1, 3], dtype="timedelta64[D]")
>>> np.mean(b)
np.timedelta64(2,'D')

指定 where 参数

>>> a = np.array([[5, 9, 13], [14, 10, 12], [11, 15, 19]])
>>> np.mean(a)
12.0
>>> np.mean(a, where=[[True], [False], [False]])
9.0