Как отфильтровать массив numpy, используя значения другого массива?

У меня есть два массива 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])
18
задан Alex Riley 10 December 2017 в 10:03
поделиться