如何从元组列表中生成组合,元组在索引0处包含字符串,在索引1处包含列表?

2024-06-02 08:24:43 发布

您现在位置:Python中文网/ 问答频道 /正文

给出tupleslist

params = ['sn', 'tp', 'v1', 'temp', 'slew']

list_tuple = [('Serial Number', [12345]),
 ('Test Points', ['TestpointA', 'TestpointC']),
 ('Voltage_1', [3.0, 3.3, 3.6, 0.0]),
 ('Temperature Setpoint', [0, 60]),
 ('Slew_1', [200, 400, 800, 1600, 3200, 6400])]

def what_i_want(test_tuple, params):
    for sn in test_tuple[0][1]:
            for tp in test_tuple[1][1]:
                for v in test_tuple[2][1]:
                    for temp in test_tuple[3][1]:
                        for slew in test_tuple[4][1]:
                            print(f'{params[0]}: ', sn)
                            print(f'{params[1]}: ', tp)
                            print(f'{params[2]}: ', v)
                            print(f'{params[3]}: ', temp)
                            print(f'{params[4]}: ', slew)
                            print('\n')

what_i_want(list_tuple, params)

产生所需输出

^{2}$

params的长度对应于list_tuple中元组的数量,这个长度可以变化。元组中每个列表的长度也可能不同(即,具有序列号的列表可以是3或4个元素,而不是1个元素)。在

我想使用递归来解包list_tuple,并通过索引正确地调用params,这样我就不必创建一个带有一堆嵌套循环的类。不过,这是我第一次使用递归,只能生成以下内容:

def poo_doo(list_tuple):
    for i in list_tuple:
        if isinstance(i, tuple):
            print(i[0])
            poo_doo(i[1])
        else:
            print(i)
            print('\n')

poo_doo(list_tuple)

产量:

Serial Number
12345


Test Points
TestpointA


TestpointC


Voltage_1
3.0


3.3
...

请帮助您使用最有效的方法生成所需的输出,最好不要使用嵌套循环。在

链接到完整类以提供更广泛的理解范围:auto_filter class


Tags: intestnumberforserialparamstemplist
1条回答
网友
1楼 · 发布于 2024-06-02 08:24:43

您可以使用itertools.product

import itertools
data = [('Serial Number', [12345]), ('Test Points', ['TestpointA', 'TestpointC']), ('Voltage_1', [3.0, 3.3, 3.6, 0.0]), ('Temperature Setpoint', [0, 60]), ('Slew_1', [200, 400, 800, 1600, 3200, 6400])]
params = ['sn', 'tp', 'v1', 'temp', 'slew']
for i in itertools.product(*[b for _, b in data]):
  print('\n'.join(f'{a}:{b}' for a, b in zip(params, i)))
  print('-'*20)

输出(前三个结果):

^{pr2}$

虽然itertools.product是(也许)这个问题的最干净的解决方案,但是可以使用一个带有生成器的简单递归函数:

def combination(d, current = []):
   if len(current) == len(data):
     yield current
   else:
     for i, a in enumerate(d):
       for c in a: 
         yield from combination(d[i+1:], current = current+[c])


for i in combination([b for _, b in data]):
  print('\n'.join(f'{a}:{b}' for a, b in zip(params, i)))
  print('-'*20)

输出(前三个结果):

sn:12345
tp:TestpointA
v1:3.0
temp:0
slew:200
          
sn:12345
tp:TestpointA
v1:3.0
temp:0
slew:400
          
sn:12345
tp:TestpointA
v1:3.0
temp:0
slew:800
          

相关问题 更多 >