Функция extract
Функция extract возвращает элементы из массива, для которых соответствующее условие истинно.
Первым параметром она принимает условие (логический массив).
Вторым параметром принимает массив, из которого извлекаются элементы.
Возвращает одномерный массив с элементами, удовлетворяющими условию.
Синтаксис
np.extract(condition, arr)
Пример
Извлечём элементы, большие 3:
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
condition = arr > 3
res = np.extract(condition, arr)
print(res)
Результат выполнения кода:
[4 5]
Пример
Извлечём элементы из двумерного массива по условию:
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
condition = arr > 4
res = np.extract(condition, arr)
print(res)
Результат выполнения кода:
[5 6 7 8 9]
Пример
Использование extract с несколькими условиями:
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
condition = (arr > 2) & (arr < 5)
res = np.extract(condition, arr)
print(res)
Результат выполнения кода:
[3 4]
Пример
Сравнение extract с индексацией по условию:
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
condition = arr % 2 == 0
res1 = np.extract(condition, arr)
res2 = arr[condition]
print("extract:", res1)
print("indexing:", res2)
Результат выполнения кода:
extract: [2 4]
indexing: [2 4]
