結果 : return diagonal elements matrix python