разделенный массив Python на неравные подмассивы

Question

Я пытаюсь разбить массив на n частей. Иногда эти части имеют одинаковый размер, иногда они имеют другой размер.

Я пытаюсь использовать:

split = np.split(list, size)

Это прекрасно работает, когда размер делится поровну на список, но в противном случае происходит сбой. Есть ли способ сделать это, чтобы «заполнить» окончательный массив дополнительными «несколькими» элементами?

Answer 1

Вы ищете np.array_split?Вот строка документа:

Split an array into multiple sub-arrays.

Please refer to the ``split`` documentation.  The only difference
between these functions is that ``array_split`` allows
`indices_or_sections` to be an integer that does *not* equally 
divide the axis.

See Also
--------
split : Split array into multiple sub-arrays of equal size.

Examples
--------
>>> x = np.arange(8.0)
>>> np.array_split(x, 3)
    [array([ 0.,  1.,  2.]), array([ 3.,  4.,  5.]), array([ 6.,  7.])]

http://docs.scipy.org/doc/numpy-1.10.0/reference/generated/numpy.array_split.html

Answer 2

def split_padded(a,n):
    padding = (-len(a))%n
    return np.split(np.concatenate((a,np.zeros(padding))),n)