时间戳

首页 > 解决方案 > 如何在 C++/犰狳中去掉不连续的索引

c++ - 如何在 C++/犰狳中去掉不连续的索引

问题描述

我正在寻找一种干净的方法来使用 C++ 的犰狳线性代数库来摆脱非连续索引。我在下面包含了一些代码，但似乎可能有更好的方法来做到这一点。任何建议表示赞赏。

以下代码适用于ind从（列）向量中删除索引a，但感觉很笨重。

for(uword k = ind.n_elem; k>0; k--){
        a.shed_row(ind(k-1));
}

有什么想法吗？

标签： c++armadillo

解决方案

这是使用模板化函数根据uvec要排除的（排序的）索引删除行的一种方法。你从那里得到缺失的索引std::set_difference，然后从那里去。

#define ARMA_USE_CXX11
#include <armadillo>
#include <iostream>

template <class T>
T drop_rows(T a, arma::uvec exclude) {
    arma::uvec full_range = arma::regspace<arma::uvec>(0, a.n_rows - 1);
    std::vector<int> diff;
    std::set_difference(full_range.begin(), full_range.end(), 
                        exclude.begin(), exclude.end(), 
                        std::inserter(diff, diff.begin()));
    T b = a.rows(arma::conv_to<arma::uvec>::from(diff));
    return b;
}

int main() {
    arma::uvec exclude = {0, 1, 4};

    arma::vec a = arma::linspace<arma::vec>(100, 500, 5);
    arma::vec b = drop_rows(a, exclude);
    std::cout << b << std::endl;

    arma::mat A = arma::mat(5, 5, arma::fill::eye);
    arma::mat B = drop_rows(A, exclude);
    std::cout << B << std::endl;

    return 0;
}

推荐阅读