dask.array.compress

dask.array.compress¶

dask.array.compress(condition, a, axis=None)[源代码]¶

沿给定轴返回数组的选择切片。

此文档字符串是从 numpy.compress 复制的。

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

当沿着给定的轴工作时，对于 condition 评估为 True 的每个索引，都会沿着该轴返回一个切片到 output 中。当处理一维数组时，compress 等同于 extract。

参数

条件布尔值的一维数组: 选择返回哪些条目的数组。如果 condition 的长度小于沿给定轴的 a 的大小，则输出将被截断为条件数组的长度。
aarray_like: 从中提取部分的数组。
轴int, 可选: 沿其进行切片操作的轴。如果为 None（默认），则在展平的数组上进行操作。
出ndarray, 可选 (Dask 不支持): 输出数组。其类型保持不变，并且它必须具有正确的形状以容纳输出。

返回

压缩数组ndarray: a 的一个副本，其中沿轴的切片在 condition 为假时被移除。

参见

take, choose, diag, diagonal, select
ndarray.compress: ndarray 中的等效方法
extract: 在处理一维数组时的等效方法
ufuncs-输出类型

示例

>>> import numpy as np  
>>> a = np.array([[1, 2], [3, 4], [5, 6]])  
>>> a  
array([[1, 2],
       [3, 4],
       [5, 6]])
>>> np.compress([0, 1], a, axis=0)  
array([[3, 4]])
>>> np.compress([False, True, True], a, axis=0)  
array([[3, 4],
       [5, 6]])
>>> np.compress([False, True], a, axis=1)  
array([[2],
       [4],
       [6]])

对展平的数组进行操作不会沿轴返回切片，而是选择元素。

>>> np.compress([False, True], a)  
array([2])

dask.array.clip

dask.array.concatenate