結果 : print diagonal elements of matrix python