用NumPy查找多项式的根

使用NumPy库可以方便地查找多项式的根，具体步骤如下：

1.导入NumPy库

import numpy as np

定义多项式的常用方法是使用NumPy中的poly1d()函数。比如，我们要定义多项式$x^2-3x+2$，可以使用以下代码：

p = np.poly1d([1, -3, 2])

其中，[1, -3, 2]表示多项式的系数，从高次到低次。

使用roots()函数可以得到多项式的所有根，语法如下：

roots = np.roots(p)

其中，p是多项式对象，roots是一个NumPy数组，包含了多项式的所有根。

以二次多项式$x^2-3x+2$为例：

import numpy as np

p = np.poly1d([1, -3, 2])
roots = np.roots(p)
print(roots)

运行以上代码，输出结果为：

[2. 1.]

可见该二次方程的两个根分别为2和1。

以三次多项式$x^3+2x^2-x-2$为例：

import numpy as np

p = np.poly1d([1, 2, -1, -2])
roots = np.roots(p)
print(roots)

运行以上代码，输出结果为：

[-2.  1. -1.]

可见该三次方程的三个根分别为-2、1和-1。

以上就是使用NumPy查找多项式的根的完整攻略，希望对您有所帮助。

Tags: Python