Элемент существует в массиве numpy? - PullRequest
0 голосов
/ 01 октября 2019

NUMP NOOB.

Как лучше всего увидеть, существует ли элемент в массиве пустышек?

Пример:

import numpy as np
a = np.array([[1, 2], [2, 3], [3, 4]])
[2, 4] in a
# This evaluates to True because there's 
# a 2 (somewhere) and a 4 (somewhere)
# but I want to match [2, 4] ONLY. So...
[2, 4] in a  # Would like this to be False
[2, 3] in a  # Would like this to be True
[3, 2] in a  # This too should be false (wrong order [3, 2] != [2, 3])

Я посмотрел на np.where() и это не похожебыть тем, что я ищу. Я получаю результат, аналогичный приведенному выше, используя np.isin([2, 4], a).

Индекс не нужен (хотя, если он подходит для поездки, все в порядке), достаточно просто логического значения.

1 Ответ

1 голос
/ 01 октября 2019

Вы ищете 2 и 4 в a, попробуйте:

[[2, 4]] in a
...