結果 : find diagonal elements matrix python