python - 测试numpy数组是否为numpy数组列表的成员，并将其从列表中删除

在测试numpy数组c是否是numpy数组CNTS列表的成员时:

import numpy as np

c = np.array([[[ 75, 763]],
              [[ 57, 763]],
              [[ 57, 749]],
              [[ 75, 749]]])

CNTS = [np.array([[[  78, 1202]],
                  [[  63, 1202]],
                  [[  63, 1187]],
                  [[  78, 1187]]]),
        np.array([[[ 75, 763]],
                  [[ 57, 763]],
                  [[ 57, 749]],
                  [[ 75, 749]]]),
        np.array([[[ 72, 742]],
                  [[ 58, 742]],
                  [[ 57, 741]],
                  [[ 57, 727]],
                  [[ 58, 726]],
                  [[ 72, 726]]]),
        np.array([[[ 66, 194]],
                  [[ 51, 194]],
                  [[ 51, 179]],
                  [[ 66, 179]]])]

print(c in CNTS)

我得到:



但是，答案很明确:c就是CNTS[1]，因此c in CNTS应该返回True!

如何正确测试numpy数组是否为numpy数组列表的成员？

删除时，也会发生同样的问题:

CNTS.remove(c)

应用程序:测试opencv轮廓(numpy数组)是否为轮廓列表的成员，例如参见Remove an opencv contour from a list of contours。

最佳答案

之所以会出现错误，是因为in本质上对bool(c == x)的每个元素x都调用CNTS。引发错误的是__bool__转换:

>>> c == CNTS[1]
array([[[ True,  True]],
       [[ True,  True]],
       [[ True,  True]],
       [[ True,  True]]])

>>> bool(_)
ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

删除同样适用，因为它会测试每个元素的相等性。

遏制

解决方案是使用np.array_equal或将all方法应用于每个比较:

any(np.array_equal(c, x) for x in CNTS)

或者

any((c == x).all() for x in CNTS)

删除

要执行删除，您对元素的索引比它的存在更感兴趣。我能想到的最快方法是使用CNTS的元素作为比较键来遍历索引:

index = next((i for i, x in enumerate(CNTS) if (c == x).all()), -1)

该选项可以很好地短路，并返回-1作为默认索引，而不是提高StopIteration。如果您喜欢此错误，可以将参数-1删除为next。如果愿意，可以将(c == x).all()替换为np.array_equal(c, x)。

现在您可以照常删除:

del CNTS[index]

关于python - 测试numpy数组是否为numpy数组列表的成员，并将其从列表中删除，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/53065245/