Question

我试图将矢量划分为2个相同大小的较小矢量。通常在R中，这将使用

完成

indices = sample(1:length(x), length(x)/2)
a = x[indices]
b = x[-indices]

在Rcpp中，我可以从RcppArmadillo复制样本函数。但是，似乎Rcpp中的子集不能处理像x[-indices]这样的事情。

Answer 1

您可以使用Node对所有索引进行随机播放，然后将前半部分提取到一个向量，将后半部分提取到另一个向量：

RcppArmadillo::sample

这将返回您的拆分列表：

// file.cpp
// [[Rcpp::depends(RcppArmadillo)]]

#include <RcppArmadilloExtensions/sample.h>

using namespace Rcpp ;

// [[Rcpp::export]]
List fxn(NumericVector x) {
  const int n = x.size();
  const int n2 = x.size()/2;

  // Randomly order indices
  NumericVector v(n);
  std::iota(v.begin(), v.end(), 0);
  NumericVector indices = RcppArmadillo::sample(v, v.size(), false);

  // Split up vectors
  NumericVector a(n2);
  NumericVector b(n - n2);
  for (int i=0; i < n2; ++i) a[i] = x[indices[i]];
  for (int i=n2; i < n; ++i) b[i-n2] = x[indices[i]];

  // Return as a list
  List ret;
  ret["a"] = a;
  ret["b"] = b;
  return ret;
}

随机将Rcpp NumericVector分成2

1 个答案: