numpy.diag() 函數返回一個新的 NumPy 數組，該數組包含輸入數組對角線上的元素
numpy.diag()
import numpy as np a = np.array([[1, 2], [3, 4]]) result = np.diag(a) print(result) # 輸出：[1 4]
在這個例子中，np.diag() 函數從二維數組 a 中提取了對角線上的元素（1 和 4），并將它們存儲在一個新的一維數組 result 中。
np.diag()
a
result
