从 RCpp 中的 NumericVector 中删除 NA

Question

我是 Rcpp 的新手(也是 C++ 的新手)。我使用此处的示例代码编写了代码 https://gist.github.com/kevinushey/4561281 来计算矩阵的行/列最大/混合，如下所示:

#include <Rcpp.h>
using namespace Rcpp;

template <class T>
inline double do_max(T& x) {
  return max(x);
}

template <class T>
inline double do_min(T& x) {
  return min(x);
}

NumericVector colMaxsCpp(NumericMatrix& x) {
  int nRows = x.nrow();
  NumericVector out = no_init(nRows);

  for (int i=0; i<nRows;i++) {
    NumericMatrix::Row tmp = x(i,_);
    out[i]=do_max(tmp);
  }

  return out;
}

NumericVector colMinsCpp(NumericMatrix& x) {
  int nRows = x.nrow();
  NumericVector out = no_init(nRows);

  for (int i=0; i<nRows;i++) {
    NumericMatrix::Row tmp = x(i,_);
    out[i]=do_min(tmp);
  }

  return out;
}

NumericVector rowMaxsCpp(NumericMatrix& x) {
  int nCols = x.ncol();
  NumericVector out = no_init(nCols);

  for (int i=0; i<nCols;i++) {
    NumericMatrix::Column tmp = x(_,i);
    out[i]=do_max(tmp);
  }

  return out;
}

NumericVector rowMinsCpp(NumericMatrix& x) {
  int nCols = x.ncol();
  NumericVector out = no_init(nCols);

  for (int i=0; i<nCols;i++) {
    NumericMatrix::Column tmp = x(_,i);
    out[i]=do_min(tmp);
  }

  return out;
}

// [[Rcpp::export]]
NumericVector Max(NumericMatrix x, int dim) {
  if (dim==1) {
    return rowMaxsCpp(x);
  } else if (dim==2) {
    return colMaxsCpp(x);
  }
}

// [[Rcpp::export]]
NumericVector Min(NumericMatrix x, int dim) {
  if (dim==1) {
    return rowMinsCpp(x);
  } else if (dim==2) {
    return colMinsCpp(x);
  }
}

我想修改代码以处理 do_min 和 do_max 函数中的 NA。 do_min 和do_max 函数使用类的模板。

我在网上读到关于 NA_REAL 和 NA_INTEGER 的内容，但这些都是特定于类的。

(a) 是否有任何通用的 NA 可用于类的 template(例如上面的示例)？

(b) 此外，是否有任何可用的函数可以使变量 x 子集以仅使用非 NA 元素，例如相当于 max(x[!is.na(x)]) 或 min(x[!is.na(x)]) 在 R ?

(c) 最后，上面的代码是用来处理数字matrix的，但即使提供了整数matrix它似乎也能正常工作(尽管输出被转换为 numeric)。为什么会这样？

Answer 1

快速的:

a) 是的，有但我不确定它们是矢量化的。但是添加了一些模板化的 NA 特征。

b) 是的，例如在

R> cppFunction('double mymax(NumericVector x) { \
                   IntegerVector x2 = wrap(na_omit(x)); \
                   return max(x2);}')
R> mymax(c(1L, 2L, NA, 4L))
[1] 4
R> 

c) Integer 总是以复制为代价“转换”为 Numeric。

@Roland 已经暗示了 na_omit。在 b) 中，我需要以某种方式帮助 wrap() 生成一个中间 SEXP 对象。