我正在考虑通过在C ++和C ++中重写来加速一些R代码。通过Rcpp整合。至少可以说,我的Cpp是生锈的:所以我会很感激任何建议。特别是,我正在寻找将函数映射到Rcpp NumericVector
的所有元素的指针。这是一个例子。
我需要按如下方式生成一个新的向量:
tail
; NumericVector
切片
到目前为止,我有这个:
// [[Rcpp::export]]
NumericVector cppAdjustProbabilities( NumericVector& currentProbs,
const int index,
const double divisor ) {
//Note index <=0, e.g. -1 means remove first element
if(index == 0) {
return(currentProbs);
} else {
NumericVector newProbs = no_init(currentProbs.size()+index);
NumericVector::iterator i = currentProbs.begin() - index;
NumericVector::iterator j = newProbs.begin();
for(; i != currentProbs.end(); ++i, ++j) {
*j=*i/divisor;
}
return(newProbs);
}
}
这有效,但我更喜欢使用“地图”方法。我查看std::transform
,但它只支持对向量元素的一元操作 - 所以我看不到如何传递除数。例如,这是无效的:
std::transform(currentProbs.begin()-index, currentProbs.end(),
newProbs.begin(), [](double val) { return (val / divisor);} );
有没有办法将divisor
纳入lambda的范围?或者另一种方式呢?
由于
答案 0 :(得分:2)
使用c++ lambda functions,您可以捕获如下值:
src1 <- 'NumericVector cppAdjustProbabilities( NumericVector& currentProbs,
const int index,
const double divisor ) {
//Note index <=0, e.g. -1 means remove first element
if(index == 0) {
return(currentProbs);
} else {
NumericVector newProbs = no_init(currentProbs.size()+index);
std::transform(currentProbs.begin()-index, currentProbs.end(),
newProbs.begin(), [&divisor](double val) { return (val / divisor);} );
// ^^^^^^^^
return(newProbs);
}
}'
Rcpp::cppFunction(src1)
currentProbs <- c(0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9)
index <- -5L
divisor <- 2.0
cppAdjustProbabilities(currentProbs, index, divisor)
#> [1] 0.30 0.35 0.40 0.45