Как проверить есть ли элемент в массиве numpy?

@aubree 

Вы можете использовать функцию numpy.isin() для проверки наличия элемента в массиве. Например:

1
2
3
4
5
6

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
value_to_check = 3
result = np.isin(value_to_check, arr)
print(result)

Результат будет True, так как элемент 3 есть в массиве arr.

Вы можете так же использовать метод numpy.ndarray.any() или numpy.ndarray.all() для проверки наличия элемента в массиве.

1
2
3
4
5

import numpy as np
arr = np.array([1, 2, 3, 4, 5])
value_to_check = 3
result = (arr == value_to_check).any()
print(result)

Результат будет True, так как элемент 3 есть в массиве arr.

1
2
3
4
5

import numpy as np
arr = np.array([1, 2, 3, 4, 5])
value_to_check = 3
result = (arr == value_to_check).all()
print(result)

Результат будет False, так как все элементы массива не равны 3.