浏览:347 时间:2025-09-18
np.diag()是NumPy库中用于处理对角矩阵的核心函数，能够提取矩阵对角线元素或构造对角矩阵，广泛应用于科学计算和数据分析领域。
numpy.diag(v, k=0)import numpy as np
arr = np.array([[1,2,3],
[4,5,6],
[7,8,9]])
diag_elements = np.diag(arr)
# 输出: [1 5 9]diag_matrix = np.diag([1, 2, 3])
# 输出:
# [[1 0 0]
# [0 2 0]
# [0 0 3]]# 提取上对角线(k=1)
np.diag(arr, k=1) # 输出: [2 6]
# 提取下对角线(k=-1)
np.diag(arr, k=-1) # 输出: [4 8]np.diag()是NumPy中处理对角线元素的重要工具，通过掌握其用法可以显著提升矩阵操作效率。结合其他NumPy函数使用，能够实现更复杂的科学计算任务。
