假设我有列表 a:
a = [0, 0, [0, 1], [0, 1, 2]]
其中 a 可以是任意长度,它的成分可以是任意长度,我想生成一个新列表,该列表可以是值的任意组合,这样对于给定的示例,我将返回:
b = [[0,0,0,0],
[0,0,0,1],
[0,0,0,2],
[0,0,1,0],
[0,0,1,1],
[0,0,1,2]]
我认为这只是循环的问题,但我将不胜感激。
谢谢
最佳答案
使用 itertools.product
:
>>> import itertools
>>> a = [0, 0, [0, 1], [0, 1, 2]]
>>> a2 = [x if isinstance(x, list) else [x] for x in a]
>>> # = [[0], [0], [0, 1], [0, 1, 2]]
>>> list(itertools.product(*a2))
[(0, 0, 0, 0), (0, 0, 0, 1), (0, 0, 0, 2), (0, 0, 1, 0), (0, 0, 1, 1), (0, 0, 1, 2)]
关于python - 使用python获取列表的排列,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/27709367/