NumPy的实用函数整理之extract

NumPy函数extract()用于根据特定条件从数组中提取特定元素。

extract()

extract(condition, arr)

其中参数：

• condition : numpy数组，元素为布尔值
• arr：与condition相同维数的numpy数组，被抽取的输入数组

函数返回值：
返回一个1维numpy数组

代码示例：
输入：

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