上代码(这里使用tril和triu都是返回array形式，还需使用mat转换回矩阵)：
而单独要提取对角线上的元素作为一个矩阵有如下两种方法：1、运用np.diag两次，再使用mat转换回矩阵：2、运用下三角矩阵减去次下三角矩阵（即对角线下的下三角阵）。
内置函数numpy.diag（a）打印矩阵的对角线元素。编程语言中预先定义的函数。嵌入到主调函数中的函数称为内置函数，又称内嵌函数。
作用是提高程序的执行效率，大多编程语言都有自己的内置函数，如javascript内置函数等。内置函数的存在极大的提升了程序员的效率和程序的阅读。
