numpy.extract() en Python

La función numpy.extract() devuelve elementos de input_array si cumplen alguna condición específica.
 

Syntax: numpy.extract(condition, array)

Parámetros:  

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.

Devoluciones : 

Array elements that satisfy the condition.

# Python Program illustrating
# numpy.compress method
 
import numpy as geek
 
array = geek.arange(10).reshape(5, 2)
print("Original array : \n", array)
 
a = geek.mod(array, 4) !=0
# This will show element status of satisfying condition
print("\nArray Condition a : \n", a)
 
# This will return elements that satisfy condition "a" condition
print("\nElements that satisfy condition a  : \n", geek.extract(a, array))
 
 
 
b = array - 4 == 1
# This will show element status of satisfying condition
print("\nArray Condition b : \n", b)
 
# This will return elements that satisfy condition "b" condition
print("\nElements that satisfy condition b  : \n", geek.extract(b, array))

Producción : 

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]

Referencias: 
https://docs.scipy.org/doc/numpy-dev/reference/generated/numpy.extract.html#numpy.extract  
Nota: 
Además, estos códigos no se ejecutarán en IDE en línea. Así que, por favor, ejecútelos en sus sistemas para explorar el funcionamiento.
Este artículo es aportado por Mohit Gupta_OMG 😀 . Si te gusta GeeksforGeeks y te gustaría contribuir, también puedes escribir un artículo usando write.geeksforgeeks.org o enviar tu artículo por correo a review-team@geeksforgeeks.org. Vea su artículo que aparece en la página principal de GeeksforGeeks y ayude a otros Geeks.
Escriba comentarios si encuentra algo incorrecto o si desea compartir más información sobre el tema tratado anteriormente.