>>379
ありがとうございます
np.split のところだけ np.array_split の方が良さそうです