在嵌套列表中映射对应的元素

时间:2019-12-05 09:07:53

标签: python list tuples zip itertools

输入:[('1', '100'), (False,), (True,), ('1', '100')] 输出:

[('1', False, True, '1'),
 ('100', False, True, '100')]

但是当我使用itertools.product时,它给了我一个完整的交叉产品

In [191]: l
Out[191]: [('1', '100'), (False,), (True,), ('1', '100')]

In [192]: list(itertools.product(*l))
Out[192]: 
[('1', False, True, '1'),
 ('1', False, True, '100'),
 ('100', False, True, '1'),
 ('100', False, True, '100')]

列表中的所有元组将具有相同的大小或具有1的大小。

我可以通过执行以下看起来不太好的

来实现此目的
In [231]: [[l[i][v] if len(l[i]) > v else l[i][0] for i in range(len(l))] for v in range(max(map(len, l)))]
Out[231]: [['1', False, True, '1'], ['100', False, True, '100']]

1 个答案:

答案 0 :(得分:0)

尝试一下:

>>> Input = [('1', '100'), (False,), (True,), ('1', '100')]
>>> Output = [sum([(i,)]+ Input[1:-1]+ [(i,)],()) for i in Input[-1]]
>>> Output
[('1', False, True, '1'), ('100', False, True, '100')]