numpy函数：diag函数

numpy.diag()是一个用于返回矩阵对角线元素或创建对角阵（diagonal array）的函数。下面通过例子来说明问题。

例子

首先引入numpy库并创建一个对角阵，如下所示：

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

输出结果为：

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

接下来，我们可以提取二维数组x的对角元素的值，构成一个一维数组。例如：

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

输出结果为：

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

我们也可以使用第二个参数来获取其他元素。请自行观察这个值在x数组中的位置。

y  np.diag(x)
print(y)

输出结果为：

[0 4 8]

如果连续使用两个diag函数，可以得到一个对角阵，除对角线以外的元素均为零。

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

输出结果为：

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

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

输出结果为：

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

版权声明：本文内容由互联网用户自发贡献，本站不承担相关法律责任.如有侵权/违法内容,本站将立刻删除。