Numpy: Filtering array (selecting some of the values from an array)



examples/numpy/filtering_array.py
import numpy as np

animals = np.array(['Cow', 'Elephant', 'Snake', 'Camel', 'Praying Mantis'])
print(animals)

longer_than_5 = np.vectorize(lambda x: len(x) > 5)
long_animals_bool = longer_than_5(animals)
print(long_animals_bool)

long_animals = animals[long_animals_bool]
print(long_animals)

['Cow' 'Elephant' 'Snake' 'Camel' 'Praying Mantis']
[False  True False False  True]
['Elephant' 'Praying Mantis']