我是 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
)。为什么会这样?
最佳答案
快速的:
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
对象。
关于从 RCpp 中的 NumericVector 中删除 NA,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/23173008/