从 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 函数使用类的 template。

我在网上看到有关 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) 整数总是以副本为代价“转换”为 Numeric。

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