Skapa vektorer

Optimera R-kod med Rcpp

Romain François

Consulting Datactive, ThinkR

Hämta en vektor från R

C++-kod

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

Anropad från 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] ;}

Flera fall

# x är redan en numerisk vektor
extract( c(13.3, 54.2), 0 )

13.3

# x är en heltalsvektor och konverteras först till en numerisk vektor
extract( 1:10, 0 )

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

# konvertering inte möjlig: fel
extract( letters, 0 )

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

Skapa en vektor med given storlek

// [[Rcpp::export]]
NumericVector ones(int n){
    // skapa en ny numerisk vektor av storleken n
    NumericVector x(n) ;    
    // manipulera den
    for( int i=0; i<n; i++){
        x[i] = 1 ; }
    return x ;
}

Anropa ones från R:

ones(10)

1 1 1 1 1 1 1 1 1 1

Konstruktorvarianter

double value = 42.0 ;
int n = 20 ;

// skapa en numerisk vektor av storleken 20 
// där alla värden är satta till 42
NumericVector x( n, value ) ;

Given uppsättning värden

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

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

Given uppsättning värden med namn

Namnge alla värden

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

Namnge bara några värden

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

Kloning av vektorer

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

    // klona x till y
    NumericVector y = clone(x) ;

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

Nu kör vi en övning!

Optimera R-kod med Rcpp