使用 Eigen 库在 C++ 中计算多项式的根

bla*_*dan 4 c++ python matlab numpy eigen

如何使用 Eigen 库在 C++ 中获得带有系数的多项式的根?

在Python中:

>>> import numpy as np
>>> coeff = [0.708563939215852, -0.3111717537041549, -0.2151830138973625]
>>> np.roots(coeff)
array([ 0.81279407, -0.37363574])
Run Code Online (Sandbox Code Playgroud)

在Matlab中:

>> coeff = [0.708563939215852, -0.3111717537041549, -0.2151830138973625]
>> roots(coeff)
ans = 
   0.812794068532020
   -0.373635742116877
Run Code Online (Sandbox Code Playgroud)

我在 C++ 中尝试使用 Eigen Library,但收到了不同的结果:

#include <unsupported/Eigen/Polynomials>

Eigen::Vector3d coeff(0.708563939215852, -0.3111717537041549, -0.2151830138973625);
Eigen::PolynomialSolver<double, Eigen::Dynamic> solver;
solver.compute(coeff);
const Eigen::PolynomialSolver<double, Eigen::Dynamic>::RootsType &r = solver.roots();

--> r[2] = {{1.2303239390096565, 0.000}, {-2.6764034787849331, 0.000}}
Run Code Online (Sandbox Code Playgroud)

bla*_*dan 6

感谢 @rafix07 的评论,以下代码给出了与 NumPy 和 MATLAB 中相同的结果。系数的顺序必须交换。

#include <unsupported/Eigen/Polynomials>

Eigen::Vector3d coeff(-0.2151830138973625, -0.3111717537041549, 0.708563939215852);
Eigen::PolynomialSolver<double, Eigen::Dynamic> solver;
solver.compute(coeff);
const Eigen::PolynomialSolver<double, Eigen::Dynamic>::RootsType &r = solver.roots();
Run Code Online (Sandbox Code Playgroud)