仅在 numpy 中打印真正的根

Question

我有这样的东西：

coefs = [28, -36, 50, -22]
print(numpy.roots(coefs))

当然结果是：

[ 0.35770550+1.11792657j  0.35770550-1.11792657j  0.57030329+0.j ]

但是，通过使用这种方法，我如何让它只打印真正的根（如果有的话）（作为浮点数）？对于我的例子来说就是这个意思：

0.57030329

Answer 1

你可以这样做，使用 iscomplex 如下：

r = numpy.roots(coefs)

In [15]: r[~numpy.iscomplex(r)]
Out[15]: array([ 0.57030329+0.j])

您也可以使用 isreal，正如评论中指出的那样：

In [17]: r[numpy.isreal(r)]
Out[17]: array([ 0.57030329+0.j])

Answer 2

不要使用 .iscomplex() 或 .isreal()，因为 roots() 是一种数值算法，它 returns 是多项式实际根的数值近似值。这可能会导致虚假的虚部，这些虚部被上述方法解释为解决方案。

示例：

# create a polynomial with these real-valued roots:
p = numpy.poly([2,3,4,5,56,6,5,4,2,3,8,0,10])
# calculate the roots from the polynomial:
r = numpy.roots(p)
print(r) # real-valued roots, with spurious imaginary part
array([ 56.00000000 +0.00000000e+00j,  10.00000000 +0.00000000e+00j,
         8.00000000 +0.00000000e+00j,   6.00000000 +0.00000000e+00j,
         5.00009796 +0.00000000e+00j,   4.99990203 +0.00000000e+00j,
         4.00008066 +0.00000000e+00j,   3.99991935 +0.00000000e+00j,
         3.00000598 +0.00000000e+00j,   2.99999403 +0.00000000e+00j,
         2.00000000 +3.77612207e-06j,   2.00000000 -3.77612207e-06j,
         0.00000000 +0.00000000e+00j])
# using isreal() fails: many correct solutions are discarded
print(r[numpy.isreal(r)])
[ 56.00000000+0.j  10.00000000+0.j   8.00000000+0.j   6.00000000+0.j
   5.00009796+0.j   4.99990203+0.j   4.00008066+0.j   3.99991935+0.j
   3.00000598+0.j   2.99999403+0.j   0.00000000+0.j]

根据您手头的问题使用一些阈值。此外，由于您对实根感兴趣，因此只保留实部：

real_valued = r.real[abs(r.imag)<1e-5] # where I chose 1-e5 as a threshold
print(real_valued)

Answer 3

希望对您有所帮助。

 roots = np.roots(coeff);
 for i in range(len(roots)):
     if np.isreal(roots[i]):
         print(np.real(roots[i]))

仅在 numpy 中打印真正的根

Print real roots only in numpy

python

numpy

python-3.x