我想知道如何将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
}
答案 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);
}