python - 两个可迭代对象的条件连接

Question

我遇到了一个奇怪的情况，导致我感到困惑(和错误)。我已将变量重命名为更清晰，但问题是真实的。我有两个需要合并的迭代。组合应该是连词，​​不能重复。但是，如果一个元素与另一个元素“相反”，则两个元素都不应该存在于最终集合中。

概念示例:(a, b, c) COMBINE (b, c', d) -> (a, b, d) # 让 c 与 c' 相反

这是我当前的实现，部分失败是因为我试图在迭代集合时修改集合的大小:

# atom_list and atom_set are distinct not only in their type but also their contents.
for atom in atom_list:
    should_add = True
    for other_atom in atom_set:
        if atom.is_opposite(other_atom):
            # Todo: cause anti-matter/matter explosion!!
            atom_set.remove(other_atom)
            should_add = False
    if should_add:
        atom_set.add(atom)

对于如何使这个更干净(并且在不修改我正在迭代的集合的情况下工作)有什么想法吗？我觉得这个问题的一个好的解决方案不仅仅是首先复制集合......

Answer 1

正如您所说，在迭代可迭代对象时修改它不是一个好主意。为什么不创建另一组？

combined_set = set()
for atom in atom_list:
    if atom.opposite() not in atom_set:
        combined_set.add(atom)

atom_list_set = set(atom_list)
for atom in atom_set:
    if atom not in atom_list_set:
        combined_set.add(atom)

这假设存在一个 opposite() 方法，该方法返回原子的相反值。第二个for循环处理atom_set中但不在atom_list中的原子。