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。

例子

1）按数量分割数组

将数组分成 3 个子数组

import numpy as np

arr = np.arange(10)
result = np.array_split(arr, 3)
print(result)

2）按特定索引分割

将数组在索引 3 和 7 处分割。

import numpy as np
arr = np.arange(10)
result = np.array_split(arr, [3, 7])
print(result)

3）按列分割二维数组

将二维数组沿列（axis=1）分割。

import numpy as np

arr2d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
result = np.array_split(arr2d, 3, axis=1)
print(result)