Rcpp如何将IntegerVector转换为NumericVector

时间:2015-05-24 15:42:51

标签: r rcpp

我想知道如何将Rcpp IntegerVector转换为NumericVetortor样本三次而不替换数字1到5。 seq_len输出IntegerVector,样本样本只接受NumericVector

// [[Rcpp::depends(RcppArmadillo)]]
#include <RcppArmadilloExtensions/sample.h>
#include <Rcpp.h>
using namespace Rcpp;

// [[Rcpp::export]]
NumericVector follow_path(NumericMatrix X, NumericVector y) {
IntegerVector i = seq_len(5)*1.0;
NumericVector n = i; //how to convert i?
return sample(cols_int,3); //sample only takes n input
}

2 个答案:

答案 0 :(得分:8)

你在这里犯了一些错误,或者我可能误解了这个问题。

首先,sample() 确实采用整数向量,实际上它是模板化的。

其次,你根本没有使用你的论据。

这是修复版本:

// [[Rcpp::depends(RcppArmadillo)]]
#include <RcppArmadilloExtensions/sample.h>
#include <Rcpp.h>
using namespace Rcpp;

// [[Rcpp::export]]
IntegerVector sampleDemo(IntegerVector iv) {   // removed unused arguments
  IntegerVector is = RcppArmadillo::sample<IntegerVector>(iv, 3, false); 
  return is;
}

/*** R
set.seed(42)
sampleDemo(c(42L, 7L, 23L, 1007L))
*/

这是它的输出:

R> sourceCpp("/tmp/soren.cpp")

R> set.seed(42)

R> sampleDemo(c(42L, 7L, 23L, 1007L))
[1] 1007   23   42
R> 

编辑:在我写这篇文章时,你自己回答了......

答案 1 :(得分:5)

我从http://adv-r.had.co.nz/Rcpp.html#rcpp-classes学习使用

NumericVector cols_num = as<NumericVector>(someIntegerVector)

// [[Rcpp::depends(RcppArmadillo)]]
#include <RcppArmadilloExtensions/sample.h>
#include <Rcpp.h>
using namespace Rcpp;
using namespace RcppArmadillo;

// [[Rcpp::export]]
NumericVector follow_path(NumericMatrix X, IntegerVector y) {
 IntegerVector cols_int = seq_len(X.ncol());
 NumericVector cols_num = as<NumericVector>(cols_int);
 return sample(cols_num,3,false);
}