导读 在数据分析和科学计算中，NumPy是一个不可或缺的工具库。今天，我们来聊聊NumPy中的`diag`函数，它可以帮助我们轻松处理对角矩阵的相关操作...
在数据分析和科学计算中，NumPy是一个不可或缺的工具库。今天，我们来聊聊NumPy中的`diag`函数，它可以帮助我们轻松处理对角矩阵的相关操作！✨
首先，`np.diag()`函数有两种主要用法：
一是从一个二维数组中提取对角线元素；二是根据一维数组创建一个对角矩阵。👀
例如，假设我们有一个二维数组 `arr = [[1, 2], [3, 4]]`，使用 `np.diag(arr)` 可以提取出主对角线 `[1, 4]`。😄
而如果输入是一维数组 `[5, 6, 7]`，则可以生成一个对角矩阵 `[[5, 0, 0], [0, 6, 0], [0, 0, 7]]`。💡
`diag`函数不仅高效，还非常适合处理高维数据运算。无论是数据分析还是机器学习建模，它都能为你节省大量时间。📊📈
掌握这个小技巧，你的NumPy技能又上了一个台阶！快去试试吧！🚀
Python NumPy 编程技巧 数据分析