np.array_split return code example

Example: numpy split

"1-D Array For example, [2, 3] would, for axis=0, result in

ary[:2]

ary[2:3]

ary[3:]