gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-11-28 20:52:53 29 4
gpt4 key购买 nike

我有任意(但统一)数字列表的任意列表。 (它们是 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:])

谢谢。

最佳答案

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

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

关于没有isinstance的列表列表上的python递归(不同),我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/5992494/

29 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com