dask.array.extract

dask.array.extract¶

dask.array.extract(condition, arr)[源代码]¶

返回满足某些条件的数组元素。

此文档字符串是从 numpy.extract 复制而来的。

Dask 版本可能存在一些不一致性。

这等同于 np.compress(ravel(condition), ravel(arr))。如果 condition 是布尔值，np.extract 等同于 arr[condition]。

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

参数

条件array_like: 一个数组，其非零或 True 的条目指示要提取的 arr 的元素。
arrarray_like: 与 condition 大小相同的输入数组。

返回

提取ndarray: 从 arr 中获取 condition 为 True 的值的排名第一的数组。

参见

take, put, copyto, compress, place

示例

>>> import numpy as np  
>>> 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])

如果 condition 是布尔值：

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

dask.array.expm1

dask.array.eye