考虑以下代码:
string = "ABCD"
variations = [list(itertools.combinations(string,x)) for x in range(1,5)]
variations
它产生以下输出:
[[('A',), ('B',), ('C',), ('D',)],
[('A', 'B'), ('A', 'C'), ('A', 'D'), ('B', 'C'), ('B', 'D'), ('C', 'D')],
[('A', 'B', 'C'), ('A', 'B', 'D'), ('A', 'C', 'D'), ('B', 'C', 'D')],
[('A', 'B', 'C', 'D')]]
但我想要的结果是:
[('A',),
('B',),
('C',),
('D',),
('A', 'B'),
('A', 'C'),
('A', 'D'),
('B', 'C'),
('B', 'D'),
('C', 'D'),
('A', 'B', 'C'),
('A', 'B', 'D'),
('A', 'C', 'D'),
('B', 'C', 'D'),
('A', 'B', 'C', 'D')]
换句话说,我想将子列表中的所有数据解压到一个列表中(以尽可能短的方式)
我试过用星号:
string = "ABCD"
variations = [*list(itertools.combinations(string,x)) for x in range(1,5)]
但它会引发以下错误:
SyntaxError: iterable unpacking cannot be used in comprehension
我该怎么办?(再说一遍,我想保持简洁)
只需将另一个
for
添加到您的列表理解中,该理解在combinations
上循环:输出:
可以在列表中执行嵌套迭代。像这样:
variations = [e for x in range(1,5) for e in (itertools.combinations(string,x)) ]
相关问题 更多 >
编程相关推荐