numpy.extract() in Python (original) (raw)

Last Updated : 08 Mar, 2024

The numpy.extract() function returns elements of input_array if they satisfy some specified condition.

Syntax: numpy.extract(condition, array)

Parameters :

array : Input array. User apply conditions on input_array elements 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 array filled with either True or False. Array elements are extracted from the Indices having True value.

Returns :

Array elements that satisfy the condition.

Python

import numpy as geek

array = geek.arange( 10 ).reshape( 5 , 2 )

print ( "Original array : \n" , array)

a = geek.mod(array, 4 ) ! = 0

print ( "\nArray Condition a : \n" , a)

print ( "\nElements that satisfy condition a : \n" , geek.extract(a, array))

b = array - 4 = = 1

print ( "\nArray Condition b : \n" , b)

print ( "\nElements that satisfy condition b : \n" , geek.extract(b, array))

Output :

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

Array Condition a : [[False True] [ True True] [False True] [ True True] [False True]]

Elements that satisfy condition a : [1 2 3 5 6 7 9]

Array Condition b : [[False False] [False False] [False True] [False False] [False False]]

Elements that satisfy condition b : [5]

Note :
Also, these codes won’t run on online IDE’s. So please, run them on your systems to explore the working.