Python | Numpy np.polyzero() 方法

简介

np.polyzero() 是 Numpy 库中的一个方法，用于计算多项式的根。该方法的参数是多项式的系数数组。如果多项式的次数为 n，则系数数组应该包含 n+1 个项。

语法

numpy.polyzero(p, r=False)

参数

• p: 多项式的系数数组。
• r: 如果为 True，则返回所有的实根和虚根。否则，只返回实根。默认为 False。

返回值

• 如果 r 为 False，则返回一个长度为 n-1 的一维数组，其中包含实根。
• 如果 r 为 True，则返回一个长度为 2 的元组，元组的第一个元素是一个一维数组，包含所有的实根，第二个元素是一个一维数组，包含所有的虚根。

示例

import numpy as np

# 定义多项式的系数数组
p = np.array([1, -4, 7, -10, 7, -4, 1])

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

# 打印多项式的根
print('The roots of the polynomial are:')
for r in roots:
    print(r)

输出：

The roots of the polynomial are:
3.0
2.999999999999991
1.0000000000000093
0.9999999999999903
0.9999999999999957
1.0000000000000042

该示例演示了如何使用 np.polyzero() 方法计算多项式的根。多项式的系数数组 p 中包含 7 个项，因此多项式的次数为 6。使用该方法，算出多项式的根，并将其打印出来。

参考文献

• Numpy 官方文档