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


Test if a numpy array is a member of a list of numpy arrays, and remove it from the list

当测试numpy数组c是否是numpy数组CNTS列表的成员时：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
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)

我得到：

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

然而，答案相当清楚：c正是CNTS[1]的意思，所以c in CNTS应该是真的！

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

移除时也会出现同样的问题：

1
CNTS.remove(c)

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

应用程序：测试opencv轮廓(numpy数组)是否是轮廓列表的成员，例如，请参见从轮廓列表中删除opencv轮廓。

相关讨论

您得到错误是因为in基本上调用bool(c == x)对CNTS的每个元素x进行调用。是__bool__转换引起了错误：

1
2
3
4
5
6
7
8
>>> 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方法应用于每个比较：

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

或

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

去除

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

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

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

现在可以像往常一样删除：

1
del CNTS[index]
相关讨论

此解决方案可用于以下情况：

1
2
3
4
5
def arrayisin(array, list_of_arrays):
    for a in list_of_arrays:
        if np.array_equal(array, a):
            return True
    return False

此函数迭代数组列表，并针对其他数组测试相等性。所以用法是：

1
2
>>> arrayisin(c, CNTS)
True

要从列表中删除数组，可以获取数组的索引，然后使用list.pop。在函数get_index中，我们枚举数组列表，这意味着我们压缩列表的索引和列表的内容。如果有匹配项，则返回匹配项的索引。

1
2
3
4
5
6
7
8
def get_index(array, list_of_arrays):
    for j, a in enumerate(list_of_arrays):
        if np.array_equal(array, a):
            return j
    return None

idx = get_index(c, CNTS)  # 1
CNTS.pop(idx)

有关list.pophttps://docs.python.org/3/tutorial/data structures.html的文档，请参阅python数据结构教程。

相关讨论

使用del删除要删除的列表的索引。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
del CNTS[int(np.where(list(np.array_equal(row, c) for row in CNTS))[0])]

CNTS

[array([[[  78, 1202]],

        [[  63, 1202]],

        [[  63, 1187]],

        [[  78, 1187]]]), array([[[ 72, 742]],

        [[ 58, 742]],

        [[ 57, 741]],

        [[ 57, 727]],

        [[ 58, 726]],

        [[ 72, 726]]]), array([[[ 66, 194]],

        [[ 51, 194]],

        [[ 51, 179]],

        [[ 66, 179]]])]
相关讨论