numpy.compress¶

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

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

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

参数

condition一维布尔数组: 选择要返回的项的数组。如果len（条件）小于 a 沿着给定的轴，然后输出被截断为条件数组的长度。
aarray_like: 从中提取零件的数组。
axis可选的: 沿其切片的轴。如果没有（默认），则使用扁平数组。
outndarray，可选: 输出数组。它的类型被保留，并且必须具有正确的形状才能保存输出。

返回

compressed_array恩达雷: 一份 a 没有沿着轴的切片 condition 是假的。

参见

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

numpy.choose numpy.diagonal