python

python diag函数的作用是什么

小樊
87
2024-09-06 14:55:34
栏目: 编程语言

numpy.diag() 函数在 NumPy 库中用于提取二维数组(矩阵)的对角线元素,或者创建一个新的对角线矩阵

  1. 提取对角线元素:当传入一个二维数组时,numpy.diag() 函数会返回该数组的主对角线元素。例如:
import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
diag_elements = np.diag(arr)
print(diag_elements)  # 输出:[1 5 9]
  1. 创建对角线矩阵:当传入一个一维数组时,numpy.diag() 函数会根据该数组的元素创建一个新的对角线矩阵。例如:
import numpy as np

arr = np.array([1, 2, 3])
diag_matrix = np.diag(arr)
print(diag_matrix)
# 输出:
# [[1 0 0]
#  [0 2 0]
#  [0 0 3]]

注意:numpy.diag() 函数还有一些其他参数,如 k,可以用来指定要提取或创建的对角线的位置。正值表示上对角线,负值表示下对角线。

0
看了该问题的人还看了