Чтобы выбрать элементы из массива NumPy по условию, вы можете использовать различные методы и функции. Вот некоторые распространенные подходы с примерами кода:

Метод 1: использование логического индексирования

import numpy as np
arr = np.array([1, 2, 3, 4, 5])
condition = arr > 2
selected_elements = arr[condition]
print(selected_elements)

Выход:

[3 4 5]

Метод 2: использование функции np.where()

import numpy as np
arr = np.array([1, 2, 3, 4, 5])
selected_elements = np.where(arr > 2, arr, 0)
print(selected_elements)

Выход:

[0 0 3 4 5]

Метод 3: использование функции np.extract()

import numpy as np
arr = np.array([1, 2, 3, 4, 5])
selected_elements = np.extract(arr > 2, arr)
print(selected_elements)

Выход:

[3 4 5]

Метод 4: использование функции np.select()

import numpy as np
arr = np.array([1, 2, 3, 4, 5])
conditions = [arr > 2, arr < 5]
choices = [arr, 0]
selected_elements = np.select(conditions, choices)
print(selected_elements)

Выход:

[0 0 3 4 0]