试图建立一个循环来改变python中3个str中的一个

Question

我的目标是创建 3 个列表。

第一个是输入：从ABCD中选择3来创建AAA，ABC...等
第二个是输出：更改每个输入的中间字母并创建一个新列表。例如：对于 AAA -> ABA、ACA、ADA。所以输入长度的3倍。
第三个是Change：我想把每一个变化都命名为c_i，比如AAA->ABA就是C1。

对于输入，

>>> lis = ["A","B","C","D"]
>>> import itertools as it
>>> inp = list(it.product(lis, repeat = 3))
>>> print(inp)
[('A', 'A', 'A'), ('A', 'A', 'B'), ... ('D', 'D', 'C'), ('D', 'D', 'D')]
>>> len(inp)
64

但我被困在如何创建输出列表上。任何想法表示赞赏！

谢谢

Answer 1

您可以使用列表推导：

import itertools

lst = ['A', 'B', 'C', 'D']

lst_input = list(itertools.product(lst, repeat=3))
lst_output = [(tup[0], x, tup[2]) for tup in lst_input for x in lst if tup[1] is not x]
lst_change = [f'C{i}' for i in range(1, len(lst_output) + 1)]

print(len(lst_input), len(lst_output), len(lst_change))
print(lst_input[:5])
print(lst_output[:5])
print(lst_change[:5])

# 64 192 192
# [('A', 'A', 'A'), ('A', 'A', 'B'), ('A', 'A', 'C'), ('A', 'A', 'D'), ('A', 'B', 'A')]
# [('A', 'B', 'A'), ('A', 'C', 'A'), ('A', 'D', 'A'), ('A', 'B', 'B'), ('A', 'C', 'B')]
# ['C1', 'C2', 'C3', 'C4', 'C5']

对于lst_input中的每个元组，中间项被所有候选字符替换，但如果替换字符与原始字符（if tup[1] is not x）相同，则替换被丢弃。