创建一组 numpy 数组

Question

我想创建一组 numpy 数组，然后高效地执行in 操作。

explored = {
       np.array([[4,9,6,1,2,5], [5,8,3,7,7,9], [2,4,1,6,3,8]]),
       np.array([[1,2,3,1,2,5], [1,2,3,7,7,9], [1,2,3,6,3,8]]),
           }

if array in explored: # where array is an np.array 
    print("the array is already in the set")
else:
    explored.append(array)

但解释器抱怨np.array 不是可散列类型。我能做些什么呢？

Answer 1

首先得到一个正则的np.array：

explored = np.array([[4,9,6,1,2,5], [5,8,3,7,7,9], [2,4,1,6,3,8]])

那么explored就是

np.array([[4,9,6,1,2,5], [5,8,3,7,7,9], [2,4,1,6,3,8]])

然后将其转换为集合。但要做到这一点，请将 np.array 中的每一行转换为一个元组，这样它就是可散列的。

explored_set = {tuple(row) for row in explored}

现在explored_set 是

{(2, 4, 1, 6, 3, 8), (4, 9, 6, 1, 2, 5), (5, 8, 3, 7, 7, 9)}

现在您可以将其用于搜索：

if tuple(array) in explored_set:
    # processing here

等等。请注意，集合的顺序与 np.array 的顺序不同，这是典型的。