假设我有一个名为 的列表列表master_list,它具有以下属性:中的每个列表都master_list包含单个数字正整数的字符串中的每个列表master_list的长度从 2 到 5的master_list长度从 1 到 8我想要做的是使用每个位置的可能性列表返回一个字符串变体列表。这是 a 的示例master_list以及输出的样子:master_list = [['3', '2', '6'], ['6', '5', '3', '9'], ['9', '8', '6']]# In this case the output would contain 3*4*3 = 36 elements output = ["339","366","399","658","636","258","268","669","668","266","369","398", "256","296","259","368","638","396","238","356","659","639","666","359", "336","299","338","696","269","358","656","698","699","298","236","239"]
1 回答
潇潇雨雨
TA贡献1833条经验 获得超4个赞
'已经尝试使用嵌套的 for 循环遍历每个列表,然后我意识到我需要使用递归,因为列表的数量是可变的。但我坚持如何做到这一点。嵌套循环让我得到了上面的输出,但它本质上是硬编码的。
试试这个(它使用itertools.product):
import itertools
li = [['3', '2', '6'], ['6', '5', '3', '9'], ['9', '8', '6']]
result = [''.join(i) for i in itertools.product(*li)]
print(result)
输出:
['369', '368', '366', '359', '358', '356', '339', '338', '336', '399', '398', '396', '269', '268', '266', '259', '258', '256', '239', '238', '236', '299', '298', '296', '669', '668', '666', '659', '658', '656', '639', '638', '636', '699', '698', '696']
如果要将每个元素转换为int:
result = [int(''.join(i)) for i in itertools.product(*li)]
添加回答
举报
0/150
提交
取消