Python numpy.array_split函数方法的使用

numpy.array_split

numpy.array_split(ary, indices_or_sections, axis=0)    [source]

将一个数组拆分为多个子数组。

请参阅split文档。 这些函数之间的唯一区别是，array_split允许indexs_or_sections是一个不等分轴的整数。 对于长度为l的数组，应将其分割为成n个部分，它将返回大小为l//n + 1的l％n个子数组，其余大小为l//n。

例子

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

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