numpy.
array_split
将一个数组拆分为多个子数组。
请参阅 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])]