np.diag( )函数

np.diag(v, k=0)
用处：提取对角线元素或构造对角线数组。

• 参数

　　　　　　　　K: 对角线的位置参数，0为默认主对角线，1为主对角线偏上一个单位，-1为主对角线偏下一个单位，以此类推。

 

 

 

 

 

　　　　v: 如果v为一个二维数组（矩阵），则返回该二维数组（矩阵）的K对角线元素，即用来提取返回对角线元素的作用。

　　　　　　如果v为一个一维数组的话，则返回一个以该一维数组作为k对角线元素的矩阵，即用来生成一个对角线元素数组。