创建一组numpy数组

首先获取常规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的顺序不同,这是典型的。