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])