numpy.压缩

麻木的。compress (条件, a , axis = None , out = None )

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

沿给定轴工作时，在条件评估为 True 的每个索引的输出中返回沿该轴的切片 。当处理一维数组时，相当于.compressextract

参数：

条件一维布尔数组

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

类似数组

要从中提取一部分的数组。

轴int，可选

沿其进行切片的轴。如果无（默认），则处理展平数组。

输出ndarray，可选

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

返回：

压缩数组ndarray

a的副本，不带沿轴的切片，条件 为 false。

也可以看看

take, choose, diag, diagonal,select

ndarray.compress

ndarray 中的等效方法

extract

处理一维数组时的等效方法

输出类型确定

例子

>>> 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])