python - 使用 Python 生成两个字符串的组合

Question

我有一个字符串 str1 = "ABC"，我需要在字符串 str1 中插入 0 到 len(str1) 个“x”，除了最后一个位置。

所以在上面的例子中，我想要一个结果列表，在字符串之前或内部放置 0 到 2 个“x”，看起来像

["ABC",
 "xABC",
 "AxBC",
 "ABxC",
 "xxABC",
 "xAxBC",
 "xABxC",
 "AxxBC",
 "AxBxC",
 "ABxxC"]

我尝试了一些代码如下(我确实意识到我错过了其中的一些组合):

>>> for i in range(1, len(str1)):
...     for k in range(len(str1)):
...         str1[:k]+"x"*i+str1[k:]
... 
'xABC'
'AxBC'
'ABxC'
'xxABC'
'AxxBC'
'ABxxC'
>>> 

我不太确定如何处理另一种情况，例如两者之间的交叉。 xAxBC 等

解决这个问题的最佳方法是什么？

Answer 1

使用 itertools.combinations_with_replacement获取索引以插入 xs.:

import itertools

str1 = "ABC"
lst = list(str1)
for n in range(len(str1)):
    for idxs in itertools.combinations_with_replacement(range(len(str1)), n):
        xs = lst[:]
        for i in reversed(idxs): # Use reversed, otherwise index become invald
            xs.insert(i, 'x')
        print(''.join(xs))

输出:

ABC
xABC
AxBC
ABxC
xxABC
xAxBC
xABxC
AxxBC
AxBxC
ABxxC