numpy.split does this already:
- http://docs.scipy.org/doc/numpy/reference/generated/numpy.split.html
Examples:
>>> mylist = np.array([1,2,3,4,5,6])
split_list(mylist,2) will return a list of two lists of three elements
- [[1,2,3][4,5,6]].
>>> np.split(mylist, 2)
[array([1, 2, 3]), array([4, 5, 6])]
split_list(mylist,3) will return a list of three lists of two
elements.
>>> np.split(mylist, 3)
[array([1, 2]), array([3, 4]), array([5, 6])]
split_list(mylist,4) will return a list of two lists of two elements
and two lists of one element.
You may probably want to add an exception capture for the cases when the remainder of length(mylist)/n is not 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
