如何从Python的嵌套列表中删除列表？

Question

我正在使用Python探索概率，我想解决此类问题。我有5对夫妇，从这些夫妻中，我想提取每组三组，而这不包含一对已婚夫妇的人。

import itertools as it

married_couples = [["Mary","John"],["Homer","Marge"],["Beauty","Beast"],["Abigail","George"],["Marco","Luisa"]]

all_persons = []
for couples in married_couples:
    for person in couples:
        all_persons.append(person)

sample_space = list(it.combinations(all_persons,3))

# Better solution with generator:
sample_space = [s for t in it.combinations(married_couples, 3) for s in it.product(*t)]

现在，我想从样本空间中排除所有包含同一对已婚夫妇的人的结果，我尝试：

correct_results = sample_space.copy()
for couple in married_couples:
    for res in sample_space:
        if couple[0] in res and couple[1] in res:
                if res in correct_results:
                    correct_results.remove(res)

编辑：我还编辑并将生成器解决方案放入其中，因此您可以将此代码用于目的

Answer 1

问题之所以出现，

if couple[0] and couple[1] in res:

是因为它测试这对夫妇在res中，而是这对夫妇的第一个元素不是null，第二个元素在res中。

您应该使用：

if couple[0] in res and couple[1] in res:

Answer 2

这是一种更有效的方法：

r = [s for t in itertools.combinations(married_couples, 3) for s in itertools.product(*t)]

如果您只需要迭代此操作，而不需要它作为实际扩展列表，那么您可以创建一个生成器：

iter = (s for t in itertools.combinations(married_couples, 3) for s in itertools.product(*t))

然后您可以这样做：

for triple in iter:
    ...

两种解决方案都可以如下：有两个级别。在顶级，它调用itertools.combinations（已婚_couples，3）。这产生了原始已婚夫妇的所有三胞胎。然后，对于每个三重态，它使用iterrools.product（*t）生成所有8个组合，这些组合从三重线中的三对中获取一个组合。