Python numpy roots用法及代碼示例

本文簡要介紹 python 語言中 numpy.roots 的用法。

用法:

numpy.roots(p)

返回具有在 p 中給出的係數的多項式的根。

注意

這構成了舊多項式 API 的一部分。從版本 1.4 開始，首選在 numpy.polynomial 中定義的新多項式 API。可以在過渡指南中找到差異摘要。

rank-1 數組 p 中的值是多項式的係數。如果 p 的長度為 n+1，則多項式說明為：

p[0] * x**n + p[1] * x**(n-1) + ... + p[n-1]*x + p[n]

參數：

p： array_like

多項式係數的 Rank-1 數組。

返回：

out： ndarray

包含多項式根的數組。

拋出：

ValueError

當 p 無法轉換為 rank-1 數組時。

注意：

該算法依賴於計算伴隨矩陣 [1] 的特征值。

參考：

1：

R. A. Horn 和 C. R. Johnson，矩陣分析。英國劍橋：劍橋大學出版社，1999 年，第 146-7 頁。

例子：

>>> coeff = [3.2, 2, 1]
>>> np.roots(coeff)
array([-0.3125+0.46351241j, -0.3125-0.46351241j])