結果 : diagonal elements of matrix python