python计算矩阵对角线之和