没有isinstance的列表列表上的python递归(不同)

Question

我有任意(但统一)数字列表的任意列表。 (它们是 n 空间中 bin 的边界坐标，我想绘制其角，但这并不重要。)我想生成所有可能组合的列表。所以:[[1,2], [3,4],[5,6]] 产生 [[1,3,5],[1,3,6],[1,4,5],[1, 4,6],[2,3,5]...].

谁能帮我改进这段代码？我不喜欢 isinstance() 调用，但是当第一个 arg (pos) 是数字列表而不是列表时，我想不出一种更像 python 的方法来在第一次传递时附加元素列表。

def recurse(pos, vals):
    out = []
    for p in pos:
        pl = p if isinstance(p,list) else [p]
        for x in vals[0]:
            out.append(pl + [x])
    if vals[1:]:
        return recurse(out, vals[1:])
    else:
        return out


a = [[1,2,3],[4,5,6],[7,8,9],[11,12,13]]

b = recurse(a[0], a[1:])

谢谢。

Answer 1

从你的例子看来你想要的只是

from itertools import product
a = [[1,2,3],[4,5,6],[7,8,9],[11,12,13]]
print list(product(*a))