如何在python中将列表拆分为给定数量的子列表[重复]

Question

可能的重复：
splitting a list of arbitrary size into only roughly N-equal parts
How do you split a list into evenly sized chunks in Python?

我需要创建一个函数，它将一个列表拆分为一个列表列表，每个列表包含相等数量的项目（或尽可能相等）。

例如

def split_lists(mainlist, splitcount):
    ....


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

split_list(mylist,2) 将返回一个包含三个元素的两个列表的列表 - [[1,2,3][4,5,6]]。

split_list(mylist,3) 将返回一个包含两个元素的三个列表的列表。

split_list(mylist,4) 将返回一个包含两个元素的两个列表和一个元素的两个列表的列表。

我不在乎哪些元素出现在哪个列表中，只是列表尽可能均匀地划分。

Answer 1

numpy.split 已经这样做了：

• http://docs.scipy.org/doc/numpy/reference/generated/numpy.split.html

例子：

>>> mylist = np.array([1,2,3,4,5,6])

split_list(mylist,2) 将返回一个包含三个元素的两个列表的列表 - [[1,2,3][4,5,6]]。

>>> np.split(mylist, 2)
[array([1, 2, 3]), array([4, 5, 6])]

split_list(mylist,3) 将返回三个列表，其中两个列表 元素。

>>> np.split(mylist, 3)
[array([1, 2]), array([3, 4]), array([5, 6])]

split_list(mylist,4) 将返回包含两个元素的两个列表的列表 和一个元素的两个列表。

您可能希望为 length(mylist)/n 的余数不为 0 的情况添加异常捕获：

>>> np.split(mylist, 4)
ValueErrorTraceback (most recent call last)
----> 1 np.split(mylist, 4)
...
ValueError: array split does not result in an equal division