使用Rcpp中随机生成的值填充向量的一部分

时间:2016-06-20 15:44:56

标签: c++ r rcpp

我正在尝试在Rcpp中编写一个Sequential Monte Carlo函数,我遇到了以下问题:

我已按以下方式创建了一个向量:

  NumericVector R_t(Part*Ttau);

我想填充矢量的Part块。应该是这样的:

for (int i=0;i<Part;i++){
        R_t[i]=runif(1,0,2);
}

第二次我想

for (int i=Part+1;i<2*Part;i++){
            R_t[i]=runif(1,0,2);
}

但它似乎不起作用。我可以在每次迭代中用新的值替换旧值,但是每次迭代我都需要旧的值。当我尝试编译时,我收到以下错误:

cannot convert 'Rcpp::NUmericVector {aka Rcpp::Vector<14, Rcpp::PrserveStorage>}' to 'Rcpp::traits::storage_type<14>:: type {aka double}' in assignment

用尺寸为Part和Ttau的二维矩阵替换矢量会更容易吗?我想避免这最后一个选择。

很抱歉,如果已经回答了这个问题,但我没有找到任何与rcpp相近的内容

2 个答案:

答案 0 :(得分:5)

您正尝试将长度为一的向量分配给需要double的位置,因此请使用[0]访问第一个元素:runif(1,0,2)[0]。但是,您也可以使用Rcpp糖结构替换您的循环,以避免一次重复生成一个随机值:

#include <Rcpp.h>

// [[Rcpp::export]]
Rcpp::NumericVector fill_vector(R_xlen_t n, R_xlen_t m) {
    Rcpp::NumericVector res(n);
    for (R_xlen_t i = 0; i < m; i++) {
        res[i] = Rcpp::runif(1, 0, 2)[0];
    }
    return res;
}

// [[Rcpp::export]]
Rcpp::NumericVector fill_vector2(R_xlen_t n, R_xlen_t m) {
    Rcpp::NumericVector res(n);
    res[Rcpp::seq(0, m - 1)] = Rcpp::runif(m, 0, 2);
    return res;
}

/***R

set.seed(123)
fill_vector(7, 4)
#[1] 0.5751550 1.5766103 0.8179538 1.7660348 0.0000000 0.0000000 0.0000000

set.seed(123)
fill_vector2(7, 4)
#[1] 0.5751550 1.5766103 0.8179538 1.7660348 0.0000000 0.0000000 0.0000000

set.seed(123)
c(runif(4, 0, 2), rep(0, 3))
#[1] 0.5751550 1.5766103 0.8179538 1.7660348 0.0000000 0.0000000 0.0000000

*/

答案 1 :(得分:3)

RNG有两种选择:

  1. 使用Rcpp sugar通过runif(n,a,b)匹配R中的Rcpp::runif(n,a,b)(返回NumericVector
  2. 创建自己的循环以模仿runif(n,a,b),每次都来自R::runif(a,b)
  3. @nrussell演示了如何通过Rcpp::runif(n,a,b)[0]对向量进行子集化来使用1,但忽略了方法2.

    以下是如何进行方法2:

    #include <Rcpp.h>
    
    // [[Rcpp::export]]
    Rcpp::NumericVector draw_vector(int n, int m) {
      Rcpp::NumericVector res(n);
      for (int i = 0; i < m; i++) {
        res[i] = R::runif(0.0, 2.0); // Draw a single element that is a double
      }
      return res;
    }
    
    /***R
    set.seed(123)
    draw_vector(7, 4)
    */
    

    这给出了:

    [1] 0.5751550 1.5766103 0.8179538 1.7660348 0.0000000 0.0000000 0.0000000