繁体   English   中英

用python中所有可能组合的另一个列表替换一个项目列表

[英]Replace one list of items with another list in all possible combinations in python

我想用所有可能的组合中的另一个列表替换一个项目列表,没有任何重复,例如

list1 =[1,2,3] 

list2 = [4,5]
Output =
[1,2,3],
[1,2,4],
[1,2,5],
[1,4,3],
[1,5,3],
[4,2,3],
[5,2,3],
[1,4,5],
[4,2,5],
[4,5,3]

我已经尝试过带 zip 的 itertools.product 但结果并不是我想要的,我想知道是否有人知道如何做到这一点,我真的很感谢你的帮助。 :)

OP 保持 list1 的顺序,并用 list2 中的元素屏蔽它。 我创建了一个 list2 的 powerset,并用 Nones 扩展它以匹配 list1(= helper)的长度。 然后我遍历 helper 的唯一排列,用另一个数组屏蔽一个数组以隐藏 list1 的部分,其中掩码包含 not-None 值。

Powerset from: 如何获取集合的所有子集? (电源组)

from itertools import chain, combinations, permutations


def powerset(iterable):
    "powerset([1,2,3]) --> () (1,) (2,) (3,) (1,2) (1,3) (2,3) (1,2,3)"
    s = list(iterable)
    return chain.from_iterable(combinations(s, r) for r in range(len(s) + 1))


list1 = [1, 2, 3]

list2 = [4, 5]

solutions = []
for s in powerset(list2):
    # for every possibility of list2
    helper = [None] * (len(list1) - len(s))
    helper.extend(s)
    # create something like [None, None, 4]
    for perm in set(permutations(helper, len(helper))):
        # for each permutation of the helper, mask out nones
        solution = []
        for listchar, helperchar in zip(list1, perm):
            if helperchar != None:
                solution.append(helperchar)
            else:
                solution.append(listchar)
        solutions.append(solution)

print(solutions)
# [[1, 2, 3], [4, 2, 3], [1, 4, 3], [1, 2, 4], [1, 2, 5], [5, 2, 3], [1, 5, 3], [1, 5, 4], [4, 2, 5], [5, 2, 4], [5, 4, 3], [1, 4, 5], [4, 5, 3]]

这是一个解决方案:

from itertools import combinations

output = list(combinations(list1 + list2, 3))

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM