如何查找 B 中包含 A 的任何元素的所有元素，其中 A 和 B 是不同长度组合的列表？

Question

假设A = [[0, 1], [1, 2]] 存储来自{0, 1, 2, 3} 的两个2 组合，而B = [[0, 1, 2], [0, 1, 3], [0, 2, 3], [1, 2, 3]] 存储来自{0, 1, 2, 3} 的所有可能的3 组合。

如何找到B 中至少包含A 中的2 个组合之一的所有元素？

我想要的输出是C = [[0, 1, 2], [0, 1, 3], [1, 2, 3]]，其中[0, 1, 2] 包括[0, 1] 和[1, 2]，[0, 1, 3] 包括[0, 1]，[1, 2, 3] 包括[1, 2]。

谢谢！ （如果答案代码高效快捷，那就太好了，因为我需要为大规模的A和B这样做）

我尝试了以下代码：

import numpy as np
import itertools

A = [[0, 1], [1, 2]]
B = []
Cdup = []
rows = range(4)
for combo in itertools.combinations(rows, 3):
    B.append(list(combo))

print(B)

for b in B:
    for a in A:
        if a.issubset(b):
            Cdup.append(b)

C = np.unique(Cdup)

但是有一个错误说list object has no attribute issubset。

Answer 1

您可以扫描listB 的每个元素b 并检查listA 的任何元素a 是否是b 的子集，在这种情况下保留b。

编码：

[b for b in listB if any(set(a).issubset(set(b)) for a in listA)]

Answer 2

这是使用itertools.combinations、推导式和set 的解决方案：

from itertools import combinations

A = [(0, 1), (1, 2)]
B = list(combinations(range(4), 3))
C = set(b for a in A for b in B if a in set(combinations(b, len(a))))

print(C)

结果：

{(0, 1, 2), (0, 1, 3), (1, 2, 3)}

上述解决方案假定len(a) <= len(b) 其中a 和b 分别是A 和B 的元素。它所做的只是检查A 的每个元素是否是B 的每个元素的组合的一个元素，由A 的元素的len() 组合。

Answer 3

希望此代码可以成为您的方案的解决方案，如果您认为需要任何解释，请告诉我：

listA=[[0,1],[1,2]]
listB=[[0,1,2],[0,1,3],[0,2,3],[1,2,3]]
res = []

for comA in listA:

    strA = " ".join(str(e) for e in comA)

    for comB in listB:
    
        strB = " ".join(str(el) for el in comB)
    
        if strA in strB:
        
            res.append(list(comB))
            # use tuple if you want to remove duplicates easily:
            # res.append(tuple(comB))


print(res)

# if the res list contain tuples the to remove duplicates we can use:
# mylist = list(dict.fromkeys(res))
# print(mylist)