在 python 3 中使用 itertools.product 和列表

Question

我正在尝试根据给定的蛋白质序列创建一个可能的密码子列表。

基本上，我尝试创建的脚本将处理给定的字符串输入并输出输入代表的另一组字符串的可能组合。

例如，字符“F”代表“UUU”或“UUC”；字符“I”代表“AUU”、“AUC”或“AUA”。

给定输入“FI”，我尝试创建的脚本应输出： “UUUAUU”、“UUUAUC”、“UUUAUA”、“UUCAUU”、“UUCAUC”和“UUCAUA”。

我目前被这段代码困住了：

import itertools

F = ['UUU', 'UUC']
I = ['AUU', 'AUC', 'AUA']

seq, pool = 'FI', []

for i in seq:
   pool.append(eval(i))

for n in itertools.product(pool):
   print(n)

当我用 pool[0], pool[1] 替换 itertools.product 中的 pool 时，它就起作用了。但我不知道如何让它工作，以便用户可以输入超过 2 个字符的字符串（即不使其硬编码）。

预先感谢您的帮助！

Answer 1

您可以在调用 product() 时使用 *pool 来“解包”列表：

for n in itertools.product(*pool):
   print(n)

此语法将列表 pool 扩展为单独的位置参数。

Answer 2

itertools.product(pool[0],pool[1],...pool[len(pool)-1]) 相当于 itertools.product(*pool)

import itertools

F = ['UUU', 'UUC']
I = ['AUU', 'AUC', 'AUA']

pool=[F,I]

for n in itertools.product(*pool):
   print(''.join(n))

Answer 3

有时（在我看来更常见）该列表不需要打印输出：

F = ['UUU', 'UUC']
I = ['AUU', 'AUC', 'AUA']

pool=[F,I]
lista = []

for n in itertools.product(*pool):
      lista.append(''.join(n))

lista