numpy.diag() 函数在 NumPy 库中用于提取二维数组（矩阵）的对角线元素，或者创建一个新的对角线矩阵
- 提取对角线元素：当传入一个二维数组时，
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]
- 创建对角线矩阵：当传入一个一维数组时，
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，可以用来指定要提取或创建的对角线的位置。正值表示上对角线，负值表示下对角线。
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，请发送邮件至 55@qq.com 举报，一经查实，本站将立刻删除。转转请注明出处：https://www.szhjjp.com/n/1133671.html
