🌟numpy.diag()函数使用详解💡

在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()` 都能助你一臂之力！快去实践吧！🚀