相当于 Rcpp 中的 'which' 函数

我是 C++ 和 Rcpp 的新手。假设我有一个向量

t1<-c(1,2,NA,NA,3,4,1,NA,5)

我想获得 t1 的元素索引NA。我可以写：

NumericVector retIdxNA(NumericVector x) {

    // Step 1: get the positions of NA in the vector
    LogicalVector y=is_na(x);

    // Step 2: count the number of NA
    int Cnt=0;
    for (int i=0;i<x.size();i++) {
       if (y[i]) {
         Cnt++;
       }
    }

    // Step 3: create an output matrix whose size is same as that of NA
    // and return the answer
    NumericVector retIdx(Cnt);
    int Cnt1=0;
    for (int i=0;i<x.size();i++) {
       if (y[i]) {
          retIdx[Cnt1]=i+1;
          Cnt1++;
       }
    }
    return retIdx;
}

然后我得到

retIdxNA(t1)
[1] 3 4 8

我想知道：

(i) 是否有任何等价物which在 Rcpp 中？

(ii) 有什么方法可以使上述函数更短/更清晰吗？特别是，有没有简单的方法可以结合上面的步骤1、2、3？

最新版本犰狳 http://dirk.eddelbuettel.com/code/rcpp.armadillo.html具有识别有限值和非有限值索引的函数。

所以这段代码

#include <RcppArmadillo.h>
// [[Rcpp::depends(RcppArmadillo)]]

// [[Rcpp::export]]
arma::uvec whichNA(arma::vec x) {
  return arma::find_nonfinite(x);
}

/*** R
t1 <- c(1,2,NA,NA,3,4,1,NA,5)
whichNA(t1)
*/

产生您想要的答案（在 C/C++ 中对离一进行模块，因为它们是从零开始的）：

R> sourceCpp("/tmp/uday.cpp")

R> t1 <- c(1,2,NA,NA,3,4,1,NA,5)

R> whichNA(t1)
     [,1]
[1,]    2
[2,]    3
[3,]    7
R> 

如果您首先创建要子集化的序列，Rcpp 也可以做到这一点：

// [[Rcpp::export]]
Rcpp::IntegerVector which2(Rcpp::NumericVector x) {
  Rcpp::IntegerVector v = Rcpp::seq(0, x.size()-1);
  return v[Rcpp::is_na(x)];
}

添加到上面的代码会产生：

R> which2(t1)
[1] 2 3 7
R> 

逻辑子集在 Rcpp 中也有些新内容。