python - 从平面列表创建列表列表

Question

这个问题已经有答案了:

How do I split a list into equally-sized chunks? (69 个回答)

已关闭 3 年前。

我知道如何将列表列表转换为平面列表。例如，

import itertools
list_ = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
merged = list(itertools.chain.from_iterable(list_))
print(merged)
>> [1, 2, 3, 4, 5, 6, 7, 8, 9] 

还有其他可能吗？在这种情况下，对于给定的平面列表，是否可以使列表的列表包含三个元素？我想看到这样的输出

[[1, 2, 3], [4, 5, 6], [7, 8, 9]]  

对于给定的输入[1, 2, 3, 4, 5, 6, 7, 8, 9]

Answer 1

只需切片:

a = [1, 2, 3, 4, 5, 6, 7, 8, 9]
b = [a[i:i+3] for i in range(0, len(a), 3)]

print(b)  # [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

或者，如果您确实想像标签所暗示的那样使用 numpy，请按照上面的建议使用 reshape()