当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python dask.array.compress用法及代码示例


用法:

dask.array.compress(condition, a, axis=None)

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

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

可能存在与 Dask 版本的一些不一致之处。

沿给定轴工作时,沿该轴的切片在 output 中为每个索引返回,其中 condition 的计算结果为 True。在处理一维数组时,compress 等效于 extract

参数

condition一维布尔数组

选择要返回的条目的数组。如果 len(condition) 小于 a 沿给定轴的大小,则输出将被截断为条件数组的长度。

aarray_like

从中提取零件的数组。

axis整数,可选

沿其进行切片的轴。如果无(默认),则处理展平数组。

outndarray,可选(在 Dask 中不支持)

输出数组。它的类型被保留,它必须具有正确的形状来保存输出。

返回

compressed_arrayndarray

a 的副本,没有沿轴的切片,其中 condition 为假。

例子

>>> 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.org大神的英文原创作品 dask.array.compress。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。