結果 : diagonal of array python