python - 如何检查一个数组是否在Python中的另一个数组中

Question

我认为 in 对此有好处，但它在不应该返回的地方返回 true。例如:

import numpy as np

a = np.array([])

for i in range(3):
    for j in range(3):
        a = np.append(a,[i,j])
a = np.reshape(a,(9,2))
print(a)

print([[0,40]] in a)

将打印为真。我不明白为什么这样做...是因为 0 在列表中吗？我想要一些只有当整个数组都在列表中时才打印 true 的东西。

我想要我的 list

[[0,1],
[0,2]]

并且仅当恰好 [0,x](相同形状相同顺序)在其中时才返回 true。

Answer 1

你可以这样做:

([0, 40] == a).all(1).any()

第一步是计算匹配位置的二维 bool 数组。然后找到所有元素都为真的行。然后检查是否有任何行完全匹配。