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

标签 python tuples combinations

我有一个数据容器,它是元组列表的列表。传入数据容器的最小大小将是一个包含两个子列表的列表,每个子列表包含两个 (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

谢谢

最佳答案

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 的事实进行了一些简化,并为最后一个子列表添加了特殊转换

关于python - 合并备用索引组合处的 n 元组列表,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50278343/

相关文章:

python - 通过 api 从树莓派 (DHT22) 收集温度在第二次尝试时失败

Python argparse 无法识别的参数

c++ - 具有自动和结构化绑定(bind)的 Ranges V3 zip

python - 当列包含 `==` 而不是 `List` 时, Pandas 比较运算符 `Tuple` 无法按预期工作

ios - 如何从元组中删除重复项

c++ - 给定数字与重复组合的算法? C++

python - 如何按唯一值分组 pandas groupby

python - 如何在 tic tac toe 游戏中检查是否连续 3 个

algorithm - 在 2 个数组中生成所有可能的元素组合的有效方法

java - 在java中打印数字数组的所有组合