溫馨提示×

python diag函數的返回值是什么

小樊
87
2024-09-06 15:15:19
欄目: 編程語言

numpy.diag() 函數返回一個新的 NumPy 數組，該數組包含輸入數組對角線上的元素

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 中。

0 踩