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])