检查numpy数组行是否在另一个2d数组中

时间:2018-03-02 02:26:50

标签: python numpy

如何找到那些尚未在a中的numpy数组ref的行索引?

例如,对于这个例子:

ref = np.asarray([[1.1, 2.2, 3.3], [4, 5, 6]])
a = np.asarray([[0, 1, 2], [1.1, 2.2, 3.3], [4, 5, 5]])

[0, 2]的结果是正确的,因为a中的这些行不在ref中。一个可能的解决方案是迭代a中的每一行并检查它是否在ref中,但我希望那里更有效和优雅(1-liner?)方式。

注意:This answer不起作用,因为它假定数组的整数元素(在我的情况下,我使用浮点数)。

0 个答案:

没有答案