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

Пользователь

от aubree , в категории: Python , 9 месяцев назад

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

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

1 ответ

Пользователь

от jakayla , 8 месяцев назад

@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.