Partition array into k subarrays python. partition # numpy.