diag
函數(shù)是 NumPy 庫中的一個函數(shù),用于從給定的數(shù)組中提取對角線元素或創(chuàng)建一個對角線矩陣
diag
函數(shù)僅適用于二維數(shù)組(矩陣),而其他函數(shù)可能適用于任意維度的數(shù)組。diag
函數(shù)可以用于提取對角線元素或創(chuàng)建對角線矩陣,而其他函數(shù)可能具有不同的功能。diag
函數(shù)的主要用途是處理矩陣的對角線元素,而其他函數(shù)可能具有更廣泛的應用范圍。例如,假設我們有一個二維數(shù)組(矩陣):
import numpy as np
matrix = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
使用 diag
函數(shù)提取對角線元素:
diagonal_elements = np.diag(matrix)
print(diagonal_elements) # 輸出:[1 5 9]
使用 diag
函數(shù)創(chuàng)建對角線矩陣:
diagonal_matrix = np.diag([1, 2, 3])
print(diagonal_matrix)
# 輸出:
# [[1 0 0]
# [0 2 0]
# [0 0 3]]
其他函數(shù)可能包括 NumPy 中的 sum
、mean
、max
等,這些函數(shù)可以用于計算數(shù)組的和、平均值、最大值等。這些函數(shù)可以應用于任意維度的數(shù)組,而不僅僅是二維數(shù)組。