How To Make Subarray In Python