結果 : python get diagonal of 2d array