diag
函数是 NumPy 库中的一个函数,用于从给定的数组中提取对角线元素或创建一个对角线矩阵
diag
函数仅适用于二维数组(矩阵),而其他函数可能适用于任意维度的数组。diag
函数可以用于提取对角线元素或创建对角线矩阵,而其他函数可能具有不同的功能。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
等,这些函数可以用于计算数组的和、平均值、最大值等。这些函数可以应用于任意维度的数组,而不仅仅是二维数组。