結果 : diagonal sum of matrix python