How To Get Subarray From Array Python