Get Subarray From Array Python Question