numpy.diag()是一個用于返回矩陣對角線元素或創(chuàng)建對角陣（diagonal array）的函數(shù)。下面通過例子來說明問題。 例子 首先引入numpy庫并創(chuàng)建一個對角陣，如下所示： impo

numpy.diag()是一個用于返回矩陣對角線元素或創(chuàng)建對角陣（diagonal array）的函數(shù)。下面通過例子來說明問題。

例子

首先引入numpy庫并創(chuàng)建一個對角陣，如下所示：

import numpy as np
x  np.diag((1, 2, 3))
print(x)

輸出結(jié)果為：

[[1 0 0]
 [0 2 0]
 [0 0 3]]

接下來，我們可以提取二維數(shù)組x的對角元素的值，構(gòu)成一個一維數(shù)組。例如：

x  (9).reshape((3,3))
print(x)

輸出結(jié)果為：

[[0 1 2]
 [3 4 5]
 [6 7 8]]

我們也可以使用第二個參數(shù)來獲取其他元素。請自行觀察這個值在x數(shù)組中的位置。

y  np.diag(x)
print(y)

輸出結(jié)果為：

[0 4 8]

如果連續(xù)使用兩個diag函數(shù)，可以得到一個對角陣，除對角線以外的元素均為零。

x  (10,19).reshape((3,3))
print(x)

輸出結(jié)果為：

[[10 11 12]
 [13 14 15]
 [16 17 18]]

z  np.diag(np.diag(x))
print(z)

輸出結(jié)果為：

[[10  0  0]
 [ 0 14  0]
 [ 0  0 18]]