У меня есть два массива NumPy, например:
a = [1,2,3,4,5]
и массив фильтров, например:
f = [False, True, False, False, True]
len(a) == len(f)
Как я могу получить новый массив numpy с только значениями в a, где один и тот же индекс в f
равен True? В моем случае: [2, 5]
.
В соответствии с принятым решением (с разными значениями):
>>> a = numpy.array([ 1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
>>> b = numpy.array([True, False, True, False, True, False, True, False, True, False])
>>> a[b]
array([1, 3, 5, 7, 9])