NumPy.roots() 函数介绍

NumPy.roots() 函数用于计算多项式的根（或零点），返回一个数组，其中包括多项式的所有实根和虚根。

语法

numpy.roots(p)

参数说明

| 参数 | 描述 | | --- | --- | | p | 一维多项式系数数组 |

返回值

返回一个数组，包含多项式的所有实根和虚根。

示例

import numpy as np

# 使用多项式系数数组创建多项式
p = np.array([1, -5, 6])

# 计算多项式的根
roots = np.roots(p)

# 输出结果
print("多项式的根为：", roots)

运行上面的代码将输出以下结果：

多项式的根为： [3. 2.]

注意事项

• 输入多项式系数数组时，按从高到低的次幂输入系数
• 如果返回的根为复数，实部和虚部可以通过 np.real() 和 np.imag() 函数获得

总结

使用 NumPy.roots() 函数可以方便地计算多项式的根，可以用于数学和工程领域的求解问题。