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
等,這些函數可以用于計算數組的和、平均值、最大值等。這些函數可以應用于任意維度的數組,而不僅僅是二維數組。