How To Get A Subarray In Python