用于从数组中提取满足某些条件的元素
numpy.extract(condition, arr)
condition |
一个布尔数组。这个布尔数组与 arr 的大小应当是一致的 |
arr |
一个数组,从这个数组中我们会根据 condition 提取元素 |
import numpy as np
a = np.array([0, 1, 2, 3, 4, 5])
condition = a % 2 == 0
result = np.extract(condition, a)
print(result)
#[0 2 4]