numpy.array_split

numpy.array_split(ary, indices_or_sections, axis=0)

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

请参阅 split 文档。这些函数之间的唯一区别是 array_split 允许 indices_or_sections 是一个整数 not 等分轴。对于应拆分为n个节的长度为l的数组，它返回大小为l//n+1的l%n子数组和大小为l//n的其余子数组。

参见

split

将数组拆分为大小相等的多个子数组。

实例

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

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

numpy.split numpy.dsplit