字符串替换组合
问题内容:
所以我有一个字符串“ 1xxx1”,我想用一个字符替换一定数量的x(也许全都不是),假设为“
5”。我想要该字符串的所有可能的组合(…也许是排列),其中x被替换或保留为x。我希望将这些结果存储在列表中。
所以期望的结果是
>>> myList = GenerateCombinations('1xxx1', '5')
>>> print myList
['1xxx1','15xx1','155x1','15551','1x5x1','1x551','1xx51']
显然,我希望它能够处理带有任意数量x的任何长度的字符串,并且能够替换任何数字。我已经尝试使用循环和递归来解决这个问题,但毫无用处。任何帮助,将不胜感激。
问题答案:
怎么样:
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
这些可能性的笛卡尔积,并将结果结合在一起。
对于奖励积分,进行修改以接受替换字典。:^)