np.diag( )函數

np.diag(v, k=0)
用處：提取對角線元素或構造對角線數組。

• 參數

　　　　　　　　K: 對角線的位置參數，0為默認主對角線，1為主對角線偏上一個單位，-1為主對角線偏下一個單位，以此類推。

 

 

 

 

 

　　　　v: 如果v為一個二維數組（矩陣），則返回該二維數組（矩陣）的K對角線元素，即用來提取返回對角線元素的作用。

　　　　　　如果v為一個一維數組的話，則返回一個以該一維數組作為k對角線元素的矩陣，即用來生成一個對角線元素數組。