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;
}

参考

posted @ 2022-02-21 12:07  Wreng  阅读(37)  评论(0编辑  收藏  举报