python - 合并备用索引组合处的 n 元组列表

Question

我有一个数据容器，它是元组列表的列表。传入数据容器的最小大小将是一个包含两个子列表的列表，每个子列表包含两个 (x,y) 形式的元组。最大传入数据大小是一个包含 100 个子列表的列表，每个子列表包含 2000 个 (x,y) 形式的元组。但是，我试图避免将上限硬编码到函数中。请注意，子列表的元组数量会有所不同。

最终目标是获取子列表并将它们合并到一个可变大小的元组有序列表中，具体取决于子列表的数量。

示例 1 有两个子列表:

a = [[(15,1),(14,2),(13,3),(12,4),(11,5),(10,6),(9,7),(8,8),(7,9),(6,10),(5,11),(4,12),(3,13),(2,14),(1,15)],
     [(7,1),(6,2),(5,3),(4,4),(3,5),(2,6),(1,7)]]

将输出(为了清晰起见，格式化):

[(15,1,7),
 (14,2,6),
 (13,3,5),
 (12,4,4),
 (11,5,3),
 (10,6,2),
 (9,7,1),
 (8,8,None),
     ...
 (1,15,None)]

示例 2 包含三个子列表:

a = [[(7,1),(6,2),(5,3),(4,4),(3,5),(2,6),(1,7)],
    [(6,1),(5,2),(4,3),(3,4),(2,5),(1,6)],
    [(10,1),(9,2),(8,3),(7,4),(6,5),(5,6),(4,7),(3,8),(2,9),(1,10)]]

会输出

[(7,1,6,1,10),
(6,2,5,2,9),
(5,3,4,3,8),
(4,4,3,4,7),
(3,5,2,5,6),
(2,6,1,6,5),
(1,7,None,7,4),
(None,None,None,None,3)
    ...]

输出基本上遵循以下形式:

对于 n 列表 > 2:

实际上 (x_{子列表 1},y_{子列表 2},x_{子列表 2},y_{子列表 2},x _子列表n)

对于 n 列表 = 2:

(x_{子列表 1},y_{子列表 1},x_{子列表 n})

我不知道如何开始——我所有的尝试都白费了。关于如何处理它的一些想法将非常感激。

注意:我必须仅使用标准库和 python 2.7

谢谢

Answer 1

itertools 库将允许您使用生成器来完成此操作。

from itertools import chain, repeat
a = ...    
longest = max(len(l) for l in a)
padded = (chain(l, repeat((None, None), longest - len(l))) for l in a[:-1])
last = chain(((x,) for x, y in a[-1]), repeat((None,), len(a[-1])))
result = [tuple(chain.from_iterable(input_tuples)) for input_tuples in zip(*chain(padded, (last,)))]

这里我使用repeat来重复适当的None元组正确的次数。 chain 用于将两个或多个生成器串在一起。

编辑:根据所有元组长度为 2 的事实进行了一些简化，并为最后一个子列表添加了特殊转换