numpy. 提取#
- 麻木的。extract (条件, arr ) [来源] #
返回满足某些条件的数组元素。
这相当于.如果 条件是布尔值则相当于.
np.compress(ravel(condition), ravel(arr))
np.extract
arr[condition]
- 参数:
- 类似条件数组
一个数组,其非零或 True 条目指示要提取的arr元素 。
- arr类似数组
输入数组的大小与条件相同。
- 返回:
- 提取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])