itertools组合复制删除

Question

假设我们有一个简单的列表，例如a = [obj_1, obj_2, obj_3, obj_4]。 其中obj_1 是基于from shapely.geometry import Polygon 的自定义对象 现在我们想要获取所有组合的列表。

from itertools import combinations
combs = combintations(a, 2)

到目前为止工作正常。在接下来的步骤中，我们要遍历所有组合并返回所有对象的列表，其中不包括属于combination 的对象。

for combination in combs:
    if combination is correct: # pseudo
        #return a list of all unique objects (obj_x), which doesn't contain combination

我试图在梳子中跟踪组合的位置，但得到了the TypeError: 'itertools.combinations' object is not subscriptable

我还在 itertools 文档中查找了删除功能，但没有发现任何有用的东西。

任何想法都值得赞赏。提前致谢

PS： 例如列表a = [obj_1, obj_2, obj_3, obj_4]如果combination = (obj_1, obj_3)是正确的，需要返回为[obj_2, obj_4]

Answer 1

迭代combs，如果元素组合正确，则从a 中删除它们。

correct_obj = []
for o1, o2 in combs:
    if (o1, o2) is correct:
        if o1 in a:
            a.remove(o1)
        if o2 in a:
            a.remove(o2)

这假设对象是可散列的，因此可以将它们放入一个集合中。如果没有，就让它成为一个普通的列表。