基于R的Rcpp接口从Armadillo向量生成随机索引的技术问询

看你给出的代码片段，应该是在实现从Armadillo向量中随机抽取一个索引的功能，但目前代码有几个小问题需要修正，我来帮你梳理一下：

原代码的问题点 #

• arma::uvec indices;没有预先分配内存空间，直接通过indices[i] = i赋值会导致内存越界错误
• RcppArmadillo::sample函数的参数不完整，缺少了是否允许重复抽样的布尔参数

修正后的完整代码 #

先给你一个能正常运行的版本：

#include <RcppArmadillo.h>
#include <RcppArmadilloExtensions/sample.h>
using namespace Rcpp;

// [[Rcpp::depends(RcppArmadillo)]]
// [[Rcpp::plugins(cpp11)]]
// [[Rcpp::export]]
int index_rand(arma::vec& v) {
  // 先给索引向量分配和输入向量一样的长度
  arma::uvec indices(v.n_elem);
  // 填充0到n-1的索引值
  for (size_t i = 0; i < v.n_elem; ++i) {
    indices[i] = i;
  }
  // 调用sample函数：抽1个元素，不允许重复抽样
  arma::uvec sampled = RcppArmadillo::sample(indices, 1, false);
  // 返回抽样得到的索引
  return sampled[0];
}

更简洁的优化写法 #

其实不用手动循环生成索引，Armadillo提供了regspace函数可以直接生成连续的整数序列，代码会更简洁高效：

#include <RcppArmadillo.h>
#include <RcppArmadilloExtensions/sample.h>
using namespace Rcpp;

// [[Rcpp::depends(RcppArmadillo)]]
// [[Rcpp::plugins(cpp11)]]
// [[Rcpp::export]]
int index_rand(arma::vec& v) {
  // 直接生成0到v.n_elem-1的无符号整数向量
  arma::uvec indices = arma::regspace<arma::uvec>(0, v.n_elem - 1);
  // 抽样1个元素，这里false表示不重复抽样，需要重复的话改成true就行
  arma::uvec sampled = RcppArmadillo::sample(indices, 1, false);
  return sampled[0];
}

关键细节说明 #

• 初始化Armadillo容器时一定要指定大小，否则会出现未定义行为
• RcppArmadillo::sample的参数顺序是：待抽样向量、抽样数量、是否允许重复、可选的权重向量（不需要的话可以省略）
• 如果你的向量长度特别大（超过int的范围），可以把返回类型改成unsigned int，但R中通常用int类型处理索引，一般场景下int足够用

内容的提问来源于stack exchange，提问作者buzaku