結果 : check diagonal matrix python