Question

我正在尝试将数组拆分为n个部分。有时这些部分的大小相同，有时它们的大小不同。

我正在尝试使用：

split = np.split(list, size)

当大小平均分配到列表中时，这可以正常工作，但否则失败。有没有办法做到这一点，将填补＆＃39;垫最后的数组与额外的＆＃39;少数＆＃39;元素？

Answer 1

您在寻找np.array_split吗？这是docstring：

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)

Answer 3

通过将索引作为列表传递，可以将数组拆分为不相等的块例子

**x = np.arange(10)**
x
(0,1,2,3,4,5,6,7,8,9)
np.array_split(x,[4])
[array([0,1,2,3],dtype = int64),
       array([4,5,6,7,8,9],dtype = int64)**

python numpy将数组分成不等的子数组

3 个答案: