创建向量

用 Rcpp 优化 R 代码

Romain François

Consulting Datactive, ThinkR

从 R 端获取向量

C++ 代码

// [[Rcpp::export]]
double extract( NumericVector x, int i){
    return x[i] ;}

从 R 调用

x <- c(13.2, 34.1)
extract(x, 0)

x[1]

13.2

13.2

// [[Rcpp::export]]
double extract( NumericVector x, int i){
    return x[i] ;}

多种情况

# x 已是数值向量
extract( c(13.3, 54.2), 0 )

13.3

# x 是整型向量，会先强制转换为数值向量
extract( 1:10, 0 )

// [[Rcpp::export]]
double extract( NumericVector x, int i){
    return x[i] ;}

# 无法转换：报错
extract( letters, 0 )

Error in extract(letters, 0) : 
  Not compatible with requested type: [type=character; target=double].

按给定大小创建向量

// [[Rcpp::export]]
NumericVector ones(int n){
    // create a new numeric vector of size n
    NumericVector x(n) ;    
    // manipulate it
    for( int i=0; i<n; i++){
        x[i] = 1 ; }
    return x ;
}

从 R 调用 ones：

ones(10)

1 1 1 1 1 1 1 1 1 1

构造函数变体

double value = 42.0 ;
int n = 20 ;

// create a numeric vector of size 20 
// with all values set to 42
NumericVector x( n, value ) ;

给定一组值

NumericVector x = NumericVector::create( 1, 2, 3 );

CharacterVector s = CharacterVector::create( "pink", "blue" );

给定具名值集

为所有值命名

NumericVector x = NumericVector::create( 
    _["a"] = 1, _["b"] = 2, _["c"] = 3
) ;

仅为部分值命名

IntegerVector y = IntegerVector::create( 
    _["d"] = 4, 5, 6, _["f"] = 7
) ;

向量克隆

// [[Rcpp::export]]
NumericVector positives( NumericVector x ){

    // clone x into y
    NumericVector y = clone(x) ;

    for( int i=0; i< y.size(); i++){
        if( y[i] < 0 ) y[i] = 0 ;
    }
    return y ;
}

Passons à la pratique !

用 Rcpp 优化 R 代码