Возврат NA из Rcpp

Nov 14 2020

Пытаюсь вернуть NA через Rcpp. Я не понимаю, почему get_na()здесь не работает, как предлагается в этом сообщении ?

> Rcpp::cppFunction('NumericVector temp1() {
   return NumericVector::get_na();
 }')
> temp1()
Error in temp1() : negative length vectors are not allowed

Если я попробую, create()это сработает.

> Rcpp::cppFunction('NumericVector temp2() {
    return NumericVector::create(NA_REAL);
  }')
> temp2()
  NA

Ответы

4 BenjaminChristoffersen Nov 14 2020 at 16:40

Вам нужно сделать что-то вроде:

Rcpp::cppFunction('NumericVector temp1() {
  NumericVector y(1);
  y[0] = NumericVector::get_na();
  return y;
}')

temp1()
#R> [1] NA

если вы хотите использовать NumericVector::get_na(). Обратите внимание, что эта функция-член просто возвращает, NA_REALпоэтому вы предположительно получаете ошибку с помощью NumericVectorконструктора:

Rcpp::cppFunction('NumericVector temp1() {
   return NumericVector::get_na();
 }')

Вы также можете использовать то, NumericVector::createчто предлагаете. Вы также можете:

Rcpp::cppFunction('NumericVector temp2() {
  return NumericVector(1, NA_REAL);
}')

или же

Rcpp::cppFunction('double temp3() {
  return NA_REAL;
}')

Возврат NA из Rcpp

Если вы имеете дело с другими типами векторов, NumericVectorтогда эта get_naфункция может быть очень полезной. Вот пример, в котором мы возвращаем NA, но с разными типами в зависимости от ввода.

Rcpp::sourceCpp(code = '
  #include "Rcpp.h"
  using namespace Rcpp;
  
  template<int T>
  Vector<T> get_na_genric(){
    return Vector<T>(1, Vector<T>::get_na());
  }
  
  // [[Rcpp::export]]
  SEXP get_nan_vec(SEXP x) {
    switch (TYPEOF(x)) {
      case INTSXP : return get_na_genric<INTSXP >();
      case LGLSXP : return get_na_genric<LGLSXP >();
      case REALSXP: return get_na_genric<REALSXP>();
      case STRSXP : return get_na_genric<STRSXP >();
      case VECSXP : return get_na_genric<VECSXP >();
      stop("type not implemented");
    }
    
    return get_na_genric<REALSXP>();
  }')

for(x in list(integer(), logical(), numeric(), character(), 
              list())){
  out <- get_nan_vec(x)
  cat("got:\n")
  print(out)
  cat("with type ", typeof(out), "\n")
}
#R> got:
#R> [1] NA
#R> with type  integer 
#R> got:
#R> [1] NA
#R> with type  logical 
#R> got:
#R> [1] NA
#R> with type  double 
#R> got:
#R> [1] NA
#R> with type  character 
#R> got:
#R> [[1]]
#R> NULL
#R> 
#R> with type  list