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
億速云公眾號
手機網站二維碼
Copyright ? Yisu Cloud Ltd. All Rights Reserved. 2018 版權所有
廣州億速云計算有限公司粵ICP備17096448號-1 粵公網安備 44010402001142號增值電信業務經營許可證編號：B1-20181529
