結果 : find diagonal matrix python