結果 : turn array into diagonal matrix python