Получить массив с только определенными индексами - PullRequest
2 голосов
/ 17 февраля 2012

Существует ли функция (или алгоритм), которая возвращает только элементы массива в зависимости от и -функции значений двух других логических массивов.

* 1003 Е.Г. *

>>> b1 = numpy.array([False, False, True, True , True])
>>> b2 = numpy.array([True , False, True, False, True])
>>> v  = numpy.array([2    , 4    , 6   , 8,     10  ])

Тогда функция должна вернуть:

numpy.array([6, 10])

Поскольку значения 6 и 10 являются значениями, для которых оба соответствующих значения b1 и b2 имеют значение True.

Отредактировано в соответствии с принятым ответом ниже:

>>> v[b1 & b2]
array([ 6, 10])

1 Ответ

4 голосов
/ 17 февраля 2012
v[b1 & b2]

сделает свое дело.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...