从RCpp中的NumericVector中删除NA
我是Rcpp的新手(也是C++的新手)。我在这里使用示例代码编写了一个代码,用于计算矩阵的row/col max/mix,如下所示:从RCpp中的NumericVector中删除NA,r,rcpp,R,Rcpp,我是Rcpp的新手(也是C++的新手)。我在这里使用示例代码编写了一个代码,用于计算矩阵的row/col max/mix,如下所示: #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) {
#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);
}
}
#包括
使用名称空间Rcpp;
模板
内联双do_最大值(T&x){
返回最大值(x);
}
模板
内联双do_最小值(T&x){
返回最小值(x);
}
数值向量colMaxsCpp(数值矩阵&x){
int nrow=x.nrow();
NumericVector out=无初始值(nRows);
对于(inti=0;iQuick-one:
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) 整数总是以副本为代价“向上转换”为数字
@罗兰已经暗示了na_omit
。在b)中,不知何故我需要帮助wrap()
生成一个中间SEXP
对象。有一个糖函数na_omit
@Roland,太好了,谢谢。我试着用max(na_omit(x))替换max(x)
但它没有编译。您能告诉我如何将它与max(x)
一起使用吗?您使用的是哪个版本的Rcpp?如果我这样做,您的代码可以编译得很好。@Roland,使用Rcpp 0.11.1
。事实上,代码已编译,但出现了类似“第70行控件到达非无效函数末尾”的错误在警告级别不足的情况下,g++
的警告是正常的。您需要插入一个无条件返回a laNumericVector NeverReached(1);return NeverReached;
来抑制它。我可以编译并使用cppFunction('double mymax(NumericVector x){return max(na_omit(x))})
没有问题。@Roland,事实上,我的错,代码确实编译了,但它给了我两个警告。Dirk在上面的评论中给了我警告的答案