結果 : diagonal of matrix python