python - 如何在特定位置生成字符串中的字符组合？

Question

我有一个字符串列表:li = ['a', 'b', 'c', 'd']在 Python 中使用以下代码，我为列表 li 生成了所有可能的字符组合。并得到了 256 个字符串的结果。

from itertools import product

li = ['a', 'b', 'c', 'd']

for comb in product(li, repeat=4):
    print(''.join(comb))

举个例子，我知道列表中字符串的第二个和第四个位置的字符 li这是'b'和'c'。
所以结果将是一组只有 16 个字符串，即:

abac
abbc
abcc
abdc
bbac
bbbc
bbcc
bbdc
cbac
cbbc
cbcc
cbdc
dbac
dbbc
dbcc
dbdc

如何得到这个结果？有没有 Pythonic 的方法来实现这一点？
谢谢。
编辑:我想要的列表大小 li是 a 到 z 和 repeat 的值是13。当我尝试上面的代码时，编译器抛出内存错误!

Answer 1

使用 list comprehension :

from itertools import product

li = ['a', 'b', 'c', 'd']
combs = [list(x) for x in product(li, repeat=4)]
selected_combs = [comb for comb in combs if (comb[1] == 'b' and comb[3] == 'c')]
print(["".join(comb) for comb in selected_combs])
# ['abac', 'abbc', 'abcc', 'abdc', 'bbac', 'bbbc', 'bbcc', 'bbdc', 'cbac', 'cbbc', 'cbcc', 'cbdc', 'dbac', 'dbbc', 'dbcc', 'dbdc']

为了节省内存，以防您不需要所有组合 combs ，你可以简单地做:

li = ['a', 'b', 'c', 'd']
selected_combs = [comb for comb in product(li, repeat=4) if (comb[1] == 'b' and comb[3] == 'c')]
print(["".join(comb) for comb in selected_combs])