使用 itertools 的特殊类型组合

Question

我几乎完成了某人给我的一项任务，该任务最初涉及轻松使用 itertools 中的 Product() 函数。 然而，该人要求它也应该做一些不同的事情，例如：

li =

[[1, 2, 3],
[4, 5, 6]]

常规的 Product() 会给出如下结果： [1, 4], [1, 5], [1, 6], [2, 4], [2, 5], [2 , 6], [3, 4] ...

它应该做的是：

做一个常规的product()，然后，从列表中的第一个元素添加下一个项目，依此类推。完整的示例集是：

[[1, 4, 2]
[1,4,3],
[1, 5, 2],
[1,5,3],
[2,4,3],
[2,5,3],
[2,6,3]]

这种情况下应该如何使用itertools？

编辑：

如果我解释该计划的目标可能会有所帮助： 例如，用户将输入 5 行 x 6 列的数字列表。
普通的product()会产生5个数字的组合。这个人想要一个 6 位数的组合。这个“第6”数字从何而来？这将取决于他对他想要的哪一行的选择。

Answer 1

我想知道你执行的神奇计算是什么，但看起来这就是你的公式：

k = int(raw_input('From What row items should be appeared again at the end?'))
res = [l for l in product(*(li+[li[k]])) if l[k]<l[len(li)] ]

Answer 2

泛化为两个以上的子列表（映射函数将是另一种选择）

from pprint import pprint
for li in ([[1, 2, 3],
            [4, 5, 6]],

           [[1,  2,  3,  4],
            [5,  6,  7,  8],
            [9, 10, 11, 12]]
           ):
    triples= []
    prevlist=li[0]
    for nextlist in li[1:]:
        for spacing in range(1,len(prevlist)):
            triples.extend([[first,other,second]
                            for first,second in zip(prevlist,prevlist[spacing:])
                            for other in nextlist])

    pprint(sorted(triples))