Question

使用以下简化代码，我希望从N（0,1）分布模拟并返回带有模拟值的列表以及取决于模拟法线的向量（请参阅下面的代码）。问题是if-else语句根本不起作用！拜托，有人可以帮我理解问题出在哪里吗？

#include <RcppArmadillo.h>
#include <Rcpp.h>
using namespace Rcpp;

//[[Rcpp::depends(RcppArmadillo)]]

//[[Rcpp::export]]

List cond(arma::vec epsilon, IntegerVector Nsim) {
int iNsim = Nsim[0];
arma::vec ans(1);
arma::vec epsil(epsilon);
arma::vec vans  = arma::zeros(iNsim);
arma::vec vcond = arma::zeros(iNsim);
LogicalVector cond;
RNGScope scope;

for (int i=0; i<iNsim; i++) {

  ans = Rcpp::rnorm(1, 0.0, 1.0);
  vans.row(i) = ans[0];
  cond = abs(ans) >= epsil;
  if (cond) {
    vcond.row(i) = 10;
  } else {
    vcond.row(i) = -10;
  }
}

return List::create(
    _["sim"] = vans,
    _["cond"] = vcond);  
}

我在R中运行它，将其保存到file.cpp，然后保存到sourceCpp（“file.cpp”）。

Answer 1

原始代码对使用向量的位置以及使用标量的位置感到困惑。

他的身材更短，修复版本：

#include <Rcpp.h>

using namespace Rcpp;

//[[Rcpp::export]]
DataFrame cond(double epsil, int iNsim) {
  double ans;
  NumericVector vans(iNsim);
  NumericVector vcond(iNsim);
  RNGScope scope;

  for (int i=0; i<iNsim; i++) {
    ans = R::rnorm(0.0, 1.0);
    vans[i] = ans;
    if (fabs(ans) >= epsil) {
      vcond[i] = 10;
    } else {
      vcond[i] = -10;
    }
  }

  return DataFrame::create(_["sim"] = vans,
                           _["cond"] = vcond);  
}

除了使用（并传递）要使用标量的标量之外，它还会将abs()更正为fabs() - 一个常见的C / C ++问题。我也恢复了Rcpp向量 - 尽管我喜欢使用Armadillo，但这里不需要它。

以下是给定随机种子的示例输出：

 R> sourceCpp("/tmp/erlis.cpp")
 R> set.seed(1)
 R> cond(1.0, 6)   
         sim cond              
 1 -0.626454  -10
 2  0.183643  -10
 3 -0.835629  -10 
 4  1.595281   10
 5  0.329508  -10
 6 -0.820468  -10
 R>

Answer 2

以下是解决方案：

#include <RcppArmadillo.h>
#include <Rcpp.h>
using namespace Rcpp;

//[[Rcpp::depends(RcppArmadillo)]]

//[[Rcpp::export]]

List cond(arma::vec epsilon, IntegerVector Nsim) {
int iNsim = Nsim[0];
arma::vec ans(1);
arma::vec epsil(epsilon);
arma::vec vans  = arma::zeros(iNsim);
arma::vec vcond = arma::zeros(iNsim);
LogicalVector cond;
RNGScope scope;

for (int i=0; i<iNsim; i++) {

  ans = Rcpp::rnorm(1, 0.0, 1.0);
  vans.row(i) = ans[0];
    bool go_ = true;
    for(int i = 0; i < ans.size(); i++)
        if(abs(ans[i]) < epsil[0])
            go_ = false;

  if (go_) {
    vcond.row(i) = 10;
  } else {
    vcond.row(i) = -10;
  }
}

return List::create(
    _["sim"] = vans,
    _["cond"] = vcond);  
}

在Rcpp中的if-else语句中遇到麻烦

2 个答案: