如何在Rcpp :: List中访问向量的元素

时间:2013-07-01 15:08:15

标签: r rcpp

我很困惑。

以下编译和工作正常:

#include <Rcpp.h>
using namespace Rcpp;
// [[Rcpp::export]]
List test(){
    List l;
    IntegerVector v(5, NA_INTEGER);
    l.push_back(v);
    return l;
}

在R:

R) test()
[[1]]
[1] NA NA NA NA NA

但是当我尝试在列表中设置IntegerVector时:

// [[Rcpp::export]]
List test(){
    List l;
    IntegerVector v(5, NA_INTEGER);
    l.push_back(v);
    l[0][1] = 1;
    return l;
}

它无法编译:

test.cpp:121:8: error: invalid use of incomplete type 'struct SEXPREC'
C:/PROGRA~1/R/R-30~1.0/include/Rinternals.h:393:16: error: forward declaration of 'struct SEXPREC'

1 个答案:

答案 0 :(得分:16)

这是因为这一行:

l[0][1] = 1;

编译器不知道l是整数向量列表。本质上l[0]为您提供SEXP(所有R对象的泛型类型),SEXP是指向SEXPREC的不透明指针,我们无法访问定义(因此不透明)。因此,当您执行[1]时,您会尝试获取第二个SEXPREC,因此不透明度使其无法实现,并且它不是您想要的。

您必须明确提取IntegerVector,因此您可以执行以下操作:

as<IntegerVector>(l[0])[1] = 1;

v[1] = 1 ;

IntegerVector x = l[0] ; x[1] = 1 ;

所有这些选项都适用于相同的底层数据结构。

或者,如果您真的想要语法l[0][1],您可以定义表达“整数向量列表”的数据结构。这是一幅草图:

template <class T>
class ListOf {
public:

    ListOf( List data_) : data(data_){}

    T operator[](int i){
        return as<T>( data[i] ) ;
    }
    operator List(){ return data ; }

private:
    List data ;
} ;

您可以使用哪种,例如像这样:

// [[Rcpp::export]]
List test2(){
    ListOf<IntegerVector> l = List::create( IntegerVector(5, NA_INTEGER) ) ; 
    l[0][1] = 1 ;
    return l;
}

另请注意,在.push_back向量(包括列表)上使用Rcpp需要列表数据的完整副本,这可能会导致您的速度变慢。只有在没有选择时才使用调整大小功能。