結果 : iterate diagonal matrix python