NumPy参考 >例行程序 >Indexing routines > numpy.compress
numpy.
compress
(condition,a,axis = None,out = None )[源代码] ¶沿给定轴返回数组的选定切片。
沿给定轴工作时,沿着该轴的切片返回在
输出为每个索引,其中条件评估为真。处理一维数组时,compress
等效于extract
。
选择要返回的条目的数组。如果len(condition)小于给定轴上a的大小,则输出将被截断为condition数组的长度。
从中提取零件的数组。
沿其进行切片的轴。如果为None(默认),则在展平的数组上工作。
输出数组。它的类型被保留,并且必须具有正确的形状以容纳输出。
没有条件沿 为false的轴的切片的a的副本。
也可以看看
take
,choose
,diag
,diagonal
,select
ndarray.compress
ndarray中的等效方法
np.extract
处理一维数组时的等效方法
ufuncs-output-type
例子
>>> 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])