結果 : split array based on index python