結果 : how to divide an array into sub arrays python