eigen 实现 sort
使用标准库 std::sort()实现，可以用 lambda 表达式定义排序的规则。
实现
void sortVec(Eigen::ArrayXd & vec)
{
std::sort(vec.data(), vec.data()+vec.size());
}
void sortVec(const Eigen::ArrayXd & vec, Eigen::ArrayXd & dst, Eigen::ArrayXi & ind)
{
ind = Eigen::ArrayXi::LinSpaced(vec.size(), 0, vec.size()-1);
auto rule = [vec](int i, int j) -> bool{return vec(i) < vec(j); }; // lambda 表达式
std::sort(ind.data(), ind.data()+ind.size(), rule);
dst.resize(vec.size());
for (int i = 0; i < vec.size(); ++i) {
dst(i) = vec(ind(i));
}
}
测试代码
#include <iostream>
#include <Eigen/Dense>
void sortVec(Eigen::ArrayXd & vec)
{
std::sort(vec.data(), vec.data()+vec.size());
}
void sortVec(const Eigen::ArrayXd & vec, Eigen::ArrayXd & dst, Eigen::ArrayXi & ind)
{
ind = Eigen::ArrayXi::LinSpaced(vec.size(), 0, vec.size()-1);
auto rule = [vec](int i, int j) -> bool{return vec(i) < vec(j); }; // lambda 表达式
std::sort(ind.data(), ind.data()+ind.size(), rule);
dst.resize(vec.size());
for (int i = 0; i < vec.size(); ++i) {
dst(i) = vec(ind(i));
}
}
int main()
{
Eigen::ArrayXd A(4), B(4), C(4);
Eigen::ArrayXi ind(4);
A << 4, 1, 3, 2;
B = A;
std::cout << "Before sorting" << std::endl;
std::cout << "A = \n" << A << std::endl;
sortVec(A);
sortVec(B, C, ind);
std::cout << "After sorting" << std::endl;
std::cout << "A = \n" << A << std::endl;
std::cout << "C = \n" << C << std::endl;
std::cout << "ind = \n" << ind << std::endl;
return 0;
}
参考
