結果 : access diagonal of matrix python