🌟numpy.diag()函数使用详解💡
发布时间：2025-03-30 22:19:18来源：
在Python的数据分析和科学计算中，`numpy` 是不可或缺的工具库之一。而 `numpy.diag()` 函数作为其中的重要成员，主要用于创建对角矩阵或提取矩阵的对角元素。无论是初学者还是资深开发者，掌握它都能大幅提升效率！💻
首先，让我们看看如何用 `numpy.diag()` 创建一个对角矩阵。只需传入一个一维数组即可：
```python
import numpy as np
diagonal = np.array([1, 2, 3])
matrix = np.diag(diagonal)
print(matrix)
```
输出结果为：
```
[[1 0 0]
[0 2 0]
[0 0 3]]
```
✨这是生成一个主对角线为 `[1, 2, 3]` 的二维矩阵。
此外，如果你想要将非主对角线设置为特定值，可以通过 `k` 参数实现！例如：
```python
matrix_k = np.diag(diagonal, k=1) k > 0 表示上对角线
print(matrix_k)
```
输出会是：
```
[[0 1 0 0]
[0 0 2 0]
[0 0 0 3]
[0 0 0 0]]
```
📝简单又高效！是不是已经迫不及待想试试了？💪
无论是数据分析、图像处理还是机器学习建模，`numpy.diag()` 都能助你一臂之力！快去实践吧！🚀
免责声明：本答案或内容为用户上传，不代表本网观点。其原创性以及文中陈述文字和内容未经本站证实，对本文以及其中全部或者部分内容、文字的真实性、完整性、及时性本站不作任何保证或承诺，请读者仅作参考，并请自行核实相关内容。 如遇侵权请及时联系本站删除。
