Python numpy extract用法及代碼示例

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

用法:

numpy.extract(condition, arr)

返回滿足某個條件的數組元素。

這相當於np.compress(ravel(condition), ravel(arr)).如果健康)狀況是布爾值np.extract相當於arr[condition].

請注意， place 的作用與 extract 完全相反。

參數：

condition： array_like

一個數組，其非零或 True 條目指示要提取的 arr 元素。

arr： array_like

與條件大小相同的輸入數組。

返回：

extract： ndarray

在條件為 True 的情況下，將 arr 的值排列為 1 數組。

例子：

>>> arr = np.arange(12).reshape((3, 4))
>>> arr
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])
>>> condition = np.mod(arr, 3)==0
>>> condition
array([[ True, False, False,  True],
       [False, False,  True, False],
       [False,  True, False, False]])
>>> np.extract(condition, arr)
array([0, 3, 6, 9])

如果條件是布爾值：

>>> arr[condition]
array([0, 3, 6, 9])