結果 : sum of diagonal elements of a matrix in python using numpy