Python 将嵌套列表视为元素时两个numpy数组的交集

例如，我有两个数组

A= [[1,0],[2,0],[3,0]]
B=[[2,1],[2,0],[3,0]]

n

p.1d（A，B）

给了我

0,1,2,3

，但我真正想要的是

[2,0]

和

[3,0]

。在这种情况下我能做什么

这里有一种方法：

set([tuple(row) for row in A]).intersection([tuple(row) for row in B])

{(3, 0), (2, 0)}

然后，您可以将其作为

np.array

获得，如下所示：

tups = set([tuple(row) for row in A]).intersection([tuple(row) for row in B])

result = np.array([tup for tup in tups])

>>> result
array([[3, 0],
       [2, 0]])