How To Create A Subarray In Python