从python中保持位置的两个字符串生成所有可能的字符串

Question

给出两个词：

AT
TC

我想生成所有可能的组合（这里不确定组合是否适用），这些组合可以通过将这两个字符串中的一个字符替换来生成，结果是：

^{pr2}$

编辑：

我试过了：

from itertools import product
ref = "ACGT"
snp = "TGCA"
prod = product(ref,snp)
for p in prod:
    print p

但结果是：

('A', 'T')
('A', 'G')
('A', 'C')
('A', 'A')
('C', 'T')
('C', 'G')
('C', 'C')
('C', 'A')
('G', 'T')
('G', 'G')
('G', 'C')
('G', 'A')
('T', 'T')
('T', 'G')
('T', 'C')
('T', 'A')

不是我要找的。我希望得到类似（每个结果的长度应与输入相同）：

ACGT
TCGT
AGGT
ACGA
....

Answer 1

您正在查找itertools.product，您可以使用如下内容：

>>> from itertools import product
>>> my_list = [ 'AT' , 'TC']

>>> list(product(*my_list))
[('A', 'T'), ('A', 'C'), ('T', 'T'), ('T', 'C')]

要以字符串形式获取这些值，可以使用列表理解作为：

^{pr2}$

编辑（基于问题中的编辑）

对于您共享的新示例，您应该将zip与上面的list comprehension表达式和itertools.product一起使用：

>>> ref = "ACGT"
>>> snp = "TGCA"

>>> [''.join(s) for s in product(*zip(ref,snp))]
['ACGT', 'ACGA', 'ACCT', 'ACCA', 'AGGT', 'AGGA', 'AGCT', 'AGCA', 'TCGT', 'TCGA', 'TCCT', 'TCCA', 'TGGT', 'TGGA', 'TGCT', 'TGCA']