結果 : matrix diagonal addition in python