結果 : get diagonal of array python