Python numpy ma.average用法及代碼示例

本文簡要介紹 python 語言中 numpy.ma.average 的用法。

用法:

ma.average(a, axis=None, weights=None, returned=False)

返回給定軸上數組的加權平均值。

參數：

a： array_like

要平均的數據。計算中不考慮屏蔽條目。

axis： 整數，可選

沿其平均 a 的軸。如果沒有，則在展平的陣列上進行平均。

weights： 數組，可選

每個元素在平均值計算中的重要性。權重數組可以是一維的(在這種情況下，它的長度必須是a沿給定軸)或具有相同的形狀a.如果weights=None，然後所有數據在a假定權重等於 1。一維計算是：

avg = sum(a * weights) / sum(weights)

對權重的唯一限製是 sum(weights) 不能為 0。

returned： 布爾型，可選

指示元組 (result, sum of weights) 是否應作為輸出返回 (True) 或僅返回結果 (False) 的標誌。默認為假。

返回：

average, [sum_of_weights]： (元組)標量或MaskedArray

沿指定軸的平均值。返回時是True，返回一個元組，其中平均值作為第一個元素，權重之和作為第二個元素。返回類型是np.float64如果a是整數類型並且浮點數小於float64，或輸入數據類型，否則。如果返回，sum_of_weights總是float64.

例子：

>>> a = np.ma.array([1., 2., 3., 4.], mask=[False, False, True, True])
>>> np.ma.average(a, weights=[3, 1, 0, 0])
1.25

>>> x = np.ma.arange(6.).reshape(3, 2)
>>> x
masked_array(
  data=[[0., 1.],
        [2., 3.],
        [4., 5.]],
  mask=False,
  fill_value=1e+20)
>>> avg, sumweights = np.ma.average(x, axis=0, weights=[1, 2, 3],
...                                 returned=True)
>>> avg
masked_array(data=[2.6666666666666665, 3.6666666666666665],
             mask=[False, False],
       fill_value=1e+20)