仅从 Python 中的列表生成某些组合

Question

我已使用以下脚本生成所有组合：

import itertools
x = 7
my_list = range(1, 11)
for k in [x]:
   for sublist in itertools.combinations(my_list, k):
      print sublist

对于第二部分，我将从 range(1, 11) 中随机抽取 6 个元素。我们称它们为 my_second_list。 我需要生成最少数量的 my_list 组合，以便获得至少一个组合以包含来自 my_second_list 的 5 个元素。

关于如何做到这一点的任何想法？

Answer 1

import itertools

x = 7
my_list = range(1,11)

for k in [x]: # isn't this just 7?
    your_output = (combination for combination in itertools.combinations(my_list,k) if all(element in combination for element in [1,2,3,4,5]))

这很丑陋，但我就是这样做的（如果我正确理解你的问题。你试图只获得那些包含某个项目子集的组合，对吧？如果你想要所有组合在包含项目子集的第一个组合之前并包括在内，我会这样做：

accumulator = list()
subset = [1,2,3,4,5] # or whatever
for k in [x]:
    for combination in itertools.combinations(my_list,k):
        accumulator.append(combination)
        if all(el in combination for el in subset):
            break

根据您的具体用例，您可能需要考虑将subset 定义为set（例如{1,2,3,4,5}）并执行subset.issubset(set(combination))，但如果不进行一些分析，很难判断这是否更好。