Numpy diag() - Python

numpy.diag() 是一个用于创建对角矩阵，或者从矩阵的对角线元素中提取出一个向量的NumPy函数。

语法：

numpy.diag(v,k=0)

参数说明：

• v：用于创建对角矩阵的一维向量
• k：对角线的偏移量

返回值

• 若输入的是一个二维矩阵，则返回其对角线上的元素
• 若输入的是一个一维向量，则返回一个对角矩阵

示例

import numpy as np
 
# 一维向量生成对角矩阵
v = np.array([1,2,3])
d = np.diag(v)
print(d)

输出：

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

# 从二维矩阵中提取对角线上的元素
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
d = np.diag(a)
print(d)

输出：

[1 5 9]

# 偏移量不为0
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
d = np.diag(a, k=1)
print(d)

输出：

[2 6]

以上就是numpy.diag()的介绍和使用方法。