結果 : define diagonal matrix python