NumPy高级函数
获取矩阵主对角线上元素以及计算矩阵的迹