np.split(x, [3, 5, 6, 10])
split(ary, indices_or_sections, axis=0) :把一个数组从左到右按顺序切分
#ary:要切分的数组
#indices_or_sections:如果是一个整数,就用该数平均切分,如果是一个数组,为沿轴切分的位置(左开右闭)
#axis:沿着哪个维度进行切向,默认为0,横向切分。为1时,纵向切分
import numpy as np
x = np.arange(8)
y = np.split(x,[3,4,6,10])
print('y = ',y)
y1 = np.split(x,[3,5,6,10])
print('y1 = ',y1)
y2 = np.split(x,[3,5,6])
print('y2 = ',y2)
y3 = np.split(x,[3,5,6,7])
print('y3 = ',y3)
---------------------------------------
y = [array([0, 1, 2]), array([3]), array([4, 5]), array([6, 7]), array([], dtype=int32)]
y1 = [array([0, 1, 2]), array([3, 4]), array([5]), array([6, 7]), array([], dtype=int32)]
y2 = [array([0, 1, 2]), array([3, 4]), array([5]), array([6, 7])]
y3 = [array([0, 1, 2]), array([3, 4]), array([5]), array([6]), array([7])]