如何在Rcpp中获取长度为1的NumericVector子向量？类似R的drop=FALSE

要在Rcpp中得到长度为1的NumericVector子向量（对应R里x[i, drop=FALSE]的行为），可以用以下两种简洁方法，避免额外辅助变量的繁琐操作：

可行方案 #

1. 使用Range索引 #

通过Range(i,i)指定单个元素的索引范围，返回结果是长度为1的NumericVector，适配函数参数要求：

NumericVector vec = {1,2,3};
int i = 0;
NumericVector sub_vec = vec[Range(i,i)];

2. 直接构造新的NumericVector #

利用NumericVector的构造函数，直接传入长度和元素值，一步生成目标向量：

NumericVector vec = {1,2,3};
int i = 0;
NumericVector sub_vec = NumericVector(1, vec[i]);

为什么vec[i]不可行？ #

Rcpp中NumericVector的operator[]返回的是标量double值，而非子向量。当你把这个标量传给期望NumericVector类型参数的函数时，会触发错误的隐式转换，导致未定义行为（比如你示例中test3()返回0就是这个原因）。

优化后的完整示例 #

#include <Rcpp.h>
using namespace Rcpp;

NumericVector test2(NumericVector x){
  return(x);
}

// [[Rcpp::export]]
NumericVector test5(){
  NumericVector vec = {1,2,3};
  int i=0;
  // 直接构造长度为1的NumericVector
  return test2(NumericVector(1, vec[i]));
}

// [[Rcpp::export]]
NumericVector test6(){
  NumericVector vec = {1,2,3};
  int i=0;
  // 使用Range索引
  return test2(vec[Range(i,i)]);
}

运行后test5()和test6()都会返回[1] 1，符合预期。

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