python diag函数与其他函数的区别

diag 函数是 NumPy 库中的一个函数，用于从给定的数组中提取对角线元素或创建一个对角线矩阵

1. diag 函数仅适用于二维数组（矩阵），而其他函数可能适用于任意维度的数组。
2. diag 函数可以用于提取对角线元素或创建对角线矩阵，而其他函数可能具有不同的功能。
3. diag 函数的主要用途是处理矩阵的对角线元素，而其他函数可能具有更广泛的应用范围。

例如，假设我们有一个二维数组（矩阵）：

import numpy as np  matrix = np.array([[1, 2, 3],                    [4, 5, 6],                    [7, 8, 9]]) 

使用 diag 函数提取对角线元素：

diagonal_elements = np.diag(matrix) print(diagonal_elements)  # 输出：[1 5 9] 

使用 diag 函数创建对角线矩阵：

diagonal_matrix = np.diag([1, 2, 3]) print(diagonal_matrix) # 输出： # [[1 0 0] #  [0 2 0] #  [0 0 3]] 

其他函数可能包括 NumPy 中的 sum、mean、max 等，这些函数可以用于计算数组的和、平均值、最大值等。这些函数可以应用于任意维度的数组，而不仅仅是二维数组。