Python NumPy average方法用法及代码示例

例子

基本用法

考虑以下：

a = np.array([1,2,3])
np.average(a, weights=[0,2,4])



2.6667

这里，加权平均值为：

(1*0 + 2*2 + 3*4) / (0+2+4) = 2.6667

获取加权平均值的总和

要获得所用加权的总和(即 0+2+4=6 )，请设置 returned=True ：

np.average([1,2,3], weights=[0,2,4], returned=True)



(2.6666666666666665, 6.0)

计算二维数组的加权平均值

假设我们有以下二维数组：

a = np.array([[1,2],[3,4]])
a



array([[1, 2],
       [3, 4]])

所有值的加权平均值

计算所有值的加权平均值：

np.average(a, weights=[[5,6],[7,8]])



2.6923076923076925

每列的加权平均值

计算每列的加权平均值，设置 axis=0 ：

np.average(a, weights=[[5,6],[7,8]], axis=0)



array([2.16666667, 3.14285714])

每行的加权平均值

计算每行的加权平均值，设置 axis=1 ：

np.average(a, weights=[[5,6],[7,8]], axis=1)



array([1.54545455, 3.53333333])