字符串替换组合

时间:2013-02-12 20:55:49

标签: python string combinations

所以我有一个字符串'1xxx1',我想用一个字符替换一个x的特定数字(也许全部可能没有),让我们说'5'。我想要所有可能的组合(...可能是排列)的字符串,其中x被替换或保留为x。我希望这些结果存储在列表中。

所以期望的结果是

>>> myList = GenerateCombinations('1xxx1', '5')
>>> print myList
['1xxx1','15xx1','155x1','15551','1x5x1','1x551','1xx51']

显然我希望它能够处理任意长度的字符串以及任何数量的x,以及能够替换任何数字。我已经尝试使用循环和递归来解决这个问题无济于事。任何帮助将不胜感激。

1 个答案:

答案 0 :(得分:12)

怎么样:

from itertools import product

def filler(word, from_char, to_char):
    options = [(c,) if c != from_char else (from_char, to_char) for c in word]
    return (''.join(o) for o in product(*options))

给出了

>>> filler("1xxx1", "x", "5")
<generator object <genexpr> at 0x8fa798c>
>>> list(filler("1xxx1", "x", "5"))
['1xxx1', '1xx51', '1x5x1', '1x551', '15xx1', '15x51', '155x1', '15551']

(请注意,您似乎缺少15x51。) 基本上,首先我们列出源词中每个字母的每个可能目标:

>>> word = '1xxx1'
>>> from_char = 'x'
>>> to_char = '5'
>>> [(c,) if c != from_char else (from_char, to_char) for c in word]
[('1',), ('x', '5'), ('x', '5'), ('x', '5'), ('1',)]

然后我们使用itertools.product来获得这些可能性的笛卡尔积,并将结果加在一起。

对于奖励积分,请修改以接受替换词典。 :^)