numpy.ma.var#
- ma.var(self, axis=None, dtype=None, out=None, ddof=0, keepdims=<no value>, mean=<no value>) = <numpy.ma.core._frommethod object>#
计算指定轴上的方差。
返回数组元素的方差,这是衡量分布分散程度的指标。默认情况下,方差是针对扁平化数组计算的,否则是在指定轴上计算的。
- 参数:
- aarray_like
包含需要计算方差的数字的数组。如果 a 不是数组,则尝试进行转换。
- axisNone 或 int 或 int 元组,可选
计算方差的轴或轴。默认情况下,计算扁平化数组的方差。
1.7.0 版本中的新增功能。
如果这是一个 int 元组,则会在多个轴上执行方差,而不是像以前那样在一个轴或所有轴上执行。
- dtype数据类型,可选
用于计算方差的类型。对于整数类型数组,默认值为
float64
;对于浮点类型数组,它与数组类型相同。- outndarray,可选
用于放置结果的备用输出数组。它必须与预期输出具有相同的形状,但类型会根据需要进行转换。
- ddof{int, float},可选
“自由度增量”:计算中使用的除数为
N - ddof
,其中N
表示元素数量。默认情况下,ddof 为零。有关 ddof 使用的详细信息,请参阅注释。- keepdimsbool,可选
如果将此设置为 True,则减少的轴将保留在结果中,作为大小为一的维度。使用此选项,结果将正确地对输入数组进行广播。
如果传递默认值,则不会将 keepdims 传递给
var
方法的ndarray
子类的子类,但是任何非默认值都会传递。如果子类的子类方法没有实现 keepdims,则会引发任何异常。- wherebool 数组,可选
要包含在方差中的元素。有关详细信息,请参阅
reduce
。1.20.0 版本中的新增功能。
- mean类数组,可选
提供平均值以防止重新计算。平均值应具有与使用
keepdims=True
计算的平均值相同的形状。计算平均值的轴应与调用此 var 函数时使用的轴相同。1.26.0 版本中的新增功能。
- correction{int, float},可选
与 Array API 兼容的
ddof
参数的名称。一次只能提供它们中的一个。2.0.0 版本中的新增功能。
- 返回:
- variancendarray,请参阅上面的 dtype 参数
如果
out=None
,则返回一个包含方差的新数组;否则,返回对输出数组的引用。
注释
数组方差计算有几种常见变体。假设输入 a 是一个一维 NumPy 数组,
mean
作为参数提供或计算为a.mean()
,NumPy 计算数组的方差为N = len(a) d2 = abs(a - mean)**2 # abs is for complex `a` var = d2.sum() / (N - ddof) # note use of `ddof`
参数 ddof 的不同值在不同的上下文中很有用。NumPy 的默认值
ddof=0
与表达式相对应\[\frac{\sum_i{|a_i - \bar{a}|^2 }}{N}\]这在统计学领域有时被称为“总体方差”,因为它将方差的定义应用于 a,就好像 a 是所有可能观测值的完整总体一样。
许多其他库对数组的方差定义不同,例如
\[\frac{\sum_i{|a_i - \bar{a}|^2}}{N - 1}\]在统计学中,所得量有时被称为“样本方差”,因为如果 a 是来自更大总体的随机样本,则此计算提供了对总体方差的无偏估计。分母中使用 \(N-1\) 通常被称为“贝塞尔校正”,因为它校正了当使用 a 的样本均值代替总体的真实均值时引入的方差估计中的偏差(朝向较低值)。对于此量,请使用
ddof=1
。请注意,对于复数,在平方之前取绝对值,因此结果始终是实数且非负数。
对于浮点输入,方差使用与输入相同的精度计算。根据输入数据,这会导致结果不准确,特别是对于
float32
(见下面的示例)。使用dtype
关键字指定更高精度的累加器可以缓解此问题。示例
>>> import numpy as np >>> a = np.array([[1, 2], [3, 4]]) >>> np.var(a) 1.25 >>> np.var(a, axis=0) array([1., 1.]) >>> np.var(a, axis=1) array([0.25, 0.25])
在单精度中,var() 可能不准确
>>> a = np.zeros((2, 512*512), dtype=np.float32) >>> a[0, :] = 1.0 >>> a[1, :] = 0.1 >>> np.var(a) 0.20250003
在 float64 中计算方差更准确
>>> np.var(a, dtype=np.float64) 0.20249999932944759 # may vary >>> ((1-0.55)**2 + (0.1-0.55)**2)/2 0.2025
指定 where 参数
>>> a = np.array([[14, 8, 11, 10], [7, 9, 10, 11], [10, 15, 5, 10]]) >>> np.var(a) 6.833333333333333 # may vary >>> np.var(a, where=[[True], [True], [False]]) 4.0
使用 mean 关键字来节省计算时间
>>> import numpy as np >>> from timeit import timeit >>> >>> a = np.array([[14, 8, 11, 10], [7, 9, 10, 11], [10, 15, 5, 10]]) >>> mean = np.mean(a, axis=1, keepdims=True) >>> >>> g = globals() >>> n = 10000 >>> t1 = timeit("var = np.var(a, axis=1, mean=mean)", globals=g, number=n) >>> t2 = timeit("var = np.var(a, axis=1)", globals=g, number=n) >>> print(f'Percentage execution time saved {100*(t2-t1)/t2:.0f}%') Percentage execution time saved 32%