【发布时间】:2016-07-27 20:59:37
【问题描述】:
- 为什么我在
np.roots的回答与np.polynomial.polynomial.polyroots不同?
定义多项式
# a specific polynomial x**0 + x**1 + x**2 + x**3
p = [1, -2.8176255165067872, -0.97639120853458261, -0.86023870029448335]
这里有一个简洁的例子来说明差异,
import numpy as np
r1 = np.roots(p); r2 = np.polynomial.polynomial.polyroots(p)
f = lambda x: np.sum([x**i*j for i,j in enumerate(p)])
print "{:>10} {:>10}".format("roots","polyroots")
for i,j in zip(r1, r2): # test should return 0
print "{:10.5f} {:10.5f}".format(np.abs(f(i)),np.abs(f(j)))
输出显然不为零
roots polyroots
46.41221 0.00000
1.97595 0.00000
1.97595 0.00000
比较的正确案例
相比之下,Mathematica 正确地获得了根:
fn[x_] := 1.` - 2.817625516506788` x - 0.97639120853458261` x^2 - 0.8602387002944835` x^3
Roots[fn[x] == 0, x]
提供根为:
x == -0.723475 - 1.78978 I || x == -0.723475 + 1.78978 I || x == 0.311926
测试验证了这一点:
fn[-0.7234748700272414` - 1.7897835665374093` I]
-4.44089*10^-16 - 2.66454*10^-15
【问题讨论】:
-
作为旁注:
p = [1, -2.8176255165067872, 2.6797211079532777, -0.86023870029448335]为np.root提供了一个体面的答案 -
似乎认为
numpy文档不正确。np.roots需要reversed()数组作为输入 -
你说得对,
np.roots需要一个反转数组——但我认为文档在这方面是完全准确的...... -
@mgilson 你是完全正确的。仔细查看文档是正确的。我花了一个小时读错了!
标签: python numpy polynomial-math polynomials