Python二维数组实现求出3*3矩阵对角线元素的和示例