結果 : how to make diagonal matrix python