Question

我想用元素组合替换列表中的变量。具体来说：

我有这两个列表

liste1 = [1,2,3,'X','X',4]
liste2 = [5,6,7]

我想得到一个包含以下元素的列表：

[1,2,3,5,6,4]
[1,2,3,5,7,4]
[1,2,3,6,7,4]
[1,2,3,6,5,4]
[1,2,3,7,5,4]
[1,2,3,7,6,4]

有人知道怎么做吗？

Answer 1

您可以这样做：

from itertools import permutations

liste1 = [1, 2, 3, 'X', 'X', 4]
liste2 = [5, 6, 7]

def replacements(liste1, liste2):
    x_indices = [i for i, val in enumerate(liste1) if val == 'X']
    nb = len(x_indices)
    for perm in permutations(liste2, nb):
        l1 = liste1[:] # if we want to preserve the original and yield different lists
        for i, new_val in zip(x_indices, perm):
            l1[i] = new_val
        yield l1

for r in replacements(liste1, liste2):
    print(r)

输出：

[1, 2, 3, 5, 6, 4]
[1, 2, 3, 5, 7, 4]
[1, 2, 3, 6, 5, 4]
[1, 2, 3, 6, 7, 4]
[1, 2, 3, 7, 5, 4]
[1, 2, 3, 7, 6, 4]

我们首先列出出现“ X”的索引，然后生成liste2的尽可能多元素的排列。对于每个排列，我们都替换为“ X”。

用元素组合替换列表中的变量

1 个答案: