Open In App

numpy.compress() in Python

Last Updated : 08 Mar, 2024
Like Article

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

Note :
This codes won’t run on online-ID. Please run them on your systems to explore the working.

Like Article
Suggest improvement
Share your thoughts in the comments

Similar Reads