Выборка элементов вектора
Для выборки элементов вектора можно использовать вектор содержащий логические значения (выражения). Выбраны будут те элементы вектора, для которых будет
True
в векторе с логическими значениями.
Пример
import numpy as np
V = np.array([1,-2,3,-4,5])
# выбор первых двух элементов вектора
print(V[np.array((True, True, False, False, False))]) # [ 1 -2]
# выбор положительных элементов вектора
print(V[V > 0]) # [1 3 5]
# выбор четных элементов вектора
print(V[V % 2 == 0]) # [-2 -4]