python:按索引将列表拆分为n个子列表

Question

我希望这个问题不是重复的；我找到了类似的，但不完全是我需要的。

我想要一种有效的方法来将列表拆分为 n 个子列表，其中每个索引转到不同的列表，直到我们到达 nth 索引然后到达下一个 n 索引以相同的顺序进入我们已有的列表，依此类推...

例如，给定以下列表:

l = [1,1,1,2,2,2,3,3,3]
n = 3

在这种情况下，我需要将列表拆分为 3 个具有所需输出的列表:

[[1,2,3],[1,2,3],[1,2,3]]

我可以制作 n for 循环来跳过每 nth 步，但我确信有更好的方法。

Answer 1

使用zip 和列表理解

l = [1,1,1,2,2,2,3,3,3]
n = 3
print([list(i) for i in zip(*[l[i:i+n] for i in range(0, len(l), n)])])

输出:

[[1, 2, 3], [1, 2, 3], [1, 2, 3]]

注意:如果 block 不均匀，您也可以使用from itertools import izip_longest。