从Rcpp中的std向量中检测并省略NA值

#include <Rcpp.h>
using namespace Rcpp;

// [[Rcpp::export]]
NumericVector removeNA(NumericVector vec) {
  NumericVector result;
  
  for (int i = 0; i < vec.size(); i++) {
    if (!NumericVector::is_na(vec[i])) {
      result.push_back(vec[i]);
    }
  }
  
  return result;
}

# 导入Rcpp库
library(Rcpp)

# 调用C++函数
vec <- c(1, 2, NA, 4, NA, 6)
new_vec <- removeNA(vec)
print(new_vec)