我尝试将计数连接到两个迭代器的乘积,如下所示:
import itertools
it1 = itertools.product(['A', 'B'], [1, 2, 3])
it2 = itertools.count(1)
list(zip(it1, it2))
这将生成输出
[(('A', 1), 1),
(('A', 2), 2),
(('A', 3), 3),
(('B', 1), 4),
(('B', 2), 5),
(('B', 3), 6)]
然而,我想要的是
[('A', 1, 1),
('A', 2, 2),
('A', 3, 3),
('B', 1, 4),
('B', 2, 5),
('B', 3, 6)]
您可以使用列表来展平元组:
输出:
相关问题 更多 >
编程相关推荐