Subarray Of Array Python Question