測試一個numpy ndarray中是否存在列表？

Question

我想知道numpy ndarray中是否存在列表[a，b]。

我試過這個：

list1 = np.array([[1, 2], [1, 3], [2, 4]])

[1, 5] in list1

結果是：

True

似乎只檢查第一個數字的存在，導致錯誤的響應。

檢查列表是否存在的好方法是什么？

Answer 1

這是一種方式：

import numpy as np

arr = np.array([[1, 2], [1, 3], [2, 4]])

lst = np.array([1, 5])
any((lst==i).all() for i in arr)  # False

lst = np.array([1, 2])
any((lst==i).all() for i in arr)  # True

Answer 2

一種簡單的方法是使用tolist() 。

>>> list1 = np.array([[1, 2], [1, 3], [2, 4]])
>>> [1, 2] in list1.tolist()
True
>>> [1, 5] in list1.tolist()
False