尝试
numpy.array_split。
从文档中：
>>> x = np.arange(8.0)>>> np.array_split(x, 3) [array([ 0., 1., 2.]), array([ 3., 4., 5.]), array([ 6., 7.])]
与相同
numpy.split，但如果组的长度不相等，则不会引发异常。
如果块数> len（array），您将获得嵌套在内部的空白数组，以解决此问题-如果将拆分数组保存在中
a，则可以通过以下方式删除空数组：
[x for x in a if x.size > 0]
只需将其保存回去即可
a。
