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

Question

У меня есть массив numpy массивов

[array([5, 5, 5]), array([6, 6, 6])]

Однако, если я попытаюсь проверить, существует ли объект в этом массиве

[5, 5, 5] in x

Я получаю эту ошибку

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

Есть ли способ исправить это? Или я что-то не так делаю?

Answer 1

Вы проверяете, существует ли список в массиве массивов. Преобразуйте ваш список в массив, и он должен работать.

>>> x=np.array([np.array([5, 5, 5]), np.array([6, 6, 6])])
>>> np.array([5, 5, 5]) in x
True