时间戳

首页 > 解决方案 > 使用 Eigen 库计算 C++ 中多项式的根

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

问题描述

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

在 Python 中：

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

在 Matlab 中：

>> coeff = [0.708563939215852, -0.3111717537041549, -0.2151830138973625]
>> roots(coeff)
ans = 
   0.812794068532020
   -0.373635742116877

我在 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}}

标签： pythonc++matlabnumpyeigen

解决方案

感谢@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();

推荐阅读