How To Create Subarray In Python