结合不同长度的多个列表中列出的名单

Question

我有这样一个列表的列表：

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

我必须通过另一个列表等给出长度的名单组合：

list2=[1, 2, 3] 

最终结果应该是这样的：[[1]，[2,3]，[4,5,6]]。

我如何可以继续做到这一点，我使用Python 3.6

Answer 1

你可以尝试在列表理解了list2使用上list1的迭代器：

>>> list1=[[1], [2], [3], [4], [5], [6]] 
>>> list2=[1, 2, 3] 
>>> v = iter(list1) 
>>> [sum((next(v) for _ in range(a)), []) for a in list2] 
[[1], [2, 3], [4, 5, 6]] 

v是在一个迭代list1; next(v)只是返回列表中的下一个子列表。在理解中，a采用list2中的值，表示我们想要的组合列表的长度。 sum(...)实际上是从list1（通过next(v)）a子列表（通过range(a)），并将它们组合成一个列表。