Open In App
Related Articles

numpy.compress() in Python

Like Article
Save Article
Report issue
The numpy.compress() function returns selected slices of an array along mentioned axis, that satisfies an axis.
Syntax: numpy.compress(condition, array, axis = None, out = None)
Parameters :
condition : [array_like]Condition on the basis of which user extract elements. 
      Applying condition on input_array, if we print condition, it will return an arra
      filled with either True or False. Array elements are extracted from the Indices having 
      True value.
array     : Input array. User apply conditions on input_array elements
axis      : [optional, int]Indicating which slice to select. 
         By Default, work on flattened array[1-D]
out       : [optional, ndarray]Output_array with elements of input_array, 
               that satisfies condition
Return :
Copy of array with elements of input_array,
that satisfies condition and along given axis
# Python Program illustrating
# numpy.compress method
import numpy as geek
array = geek.arange(10).reshape(5, 2)
print("Original array : \n", array)
a = geek.compress([0, 1], array, axis=0)
print("\nSliced array : \n", a)
a = geek.compress([False, True], array, axis=0)
print("\nSliced array : \n", a)

Output :
Original array : 
 [[0 1]
 [2 3]
 [4 5]
 [6 7]
 [8 9]]

Sliced array : 
 [[2 3]]

Sliced array : 
 [[2 3]]
References : Note : This codes won’t run on online-ID. Please run them on your systems to explore the working. .

Last Updated : 22 Oct, 2020
Like Article
Save Article
Share your thoughts in the comments
Similar Reads