python - 如何在 Python 3 中相应地将括号放入列表中？

Question

<分区>

Answer 1

这是一种使用 iterator 的方法进行更改。

list_1 = [1,2,2]
list_2 = [1,2,3,4,5]

it = iter(list_2)
res = []
for take in list_1:
  res.append([next(it) for _ in range(take)])
print(res)  # -> [[1], [2, 3], [4, 5]]

当然，也存在列表理解版本¹:.

it = iter(list_2)
res = [[next(it) for _ in range(take)] for take in list_1]

---

最后，正如@Chris_Rands 在下面的评论中指出的那样，您可以使用 islice来自非常有用的 itertools模块。它负责上面代码的迭代器创建部分，因此更加紧凑。

import itertools


res = [list(itertools.islice(list_2, take)) for take in list_1]
print(res)  # -> [[1], [2, 3], [4, 5]]

============================================= ===========================

¹ 记得重新创建迭代器!