详解Numpy roots()（求解多项式的根）函数的作用与使用方法

Numpy 中的 roots() 函数可以用于求解多项式的根。它返回给定多项式的根，其输入参数为一维多项式系数数组，返回值为一个包含所有根的一维数组。

使用方法

numpy.roots(p)

参数：

• p：包含 N+1 个元素的一维 NumPy 数组，表示多项式每个系数。

返回值：

一维数组，包含多项式的根。

例子1

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

输出：

array([2., 1.])

这个例子计算了多项式 y = x^2 - 3x + 2 的根，并输出了数组 [2., 1.]，即此多项式的两个根。

例子2

import numpy as np
p = np.poly1d([1, 0, -1, 0])
r = np.roots(p)
print(r)

输出：

array([ 1., -1.,  0.])

这个例子计算了多项式 y = x^3 - x 的根，并输出了数组 [ 1, -1, 0]，即此多项式的三个根。