Я не понимаю этого поведения в Rcpp

Question

Я хотел бы получить доступ к значению второго уникального элемента в целочисленном векторе, используя Rcpp, но вместо этого получаю вектор нулей, равный по длине значению второго элемент в целочисленном векторе. Что я делаю не так?

require(Rcpp) 
cppFunction("NumericVector test(IntegerVector labelling1) { 
    IntegerVector lvls = unique(labelling1); 
    return(lvls[1]); 
}") 

    test(1:5) 
#[1] 0 0 

Answer 1

Существует на самом деле отдельный вопрос здесь: вы пытаетесь построить NumericVector из int и Rcpp выполняет следующие действия:

1. Целочисленное значение 2 возвращается вашей функцией,
2. A NumericVector сконструирован как NumericVector(2); т.е. NumericVector с длиной 2.

Если то, что вы действительно хотите это IntegerVector, представляющее значение по этому индексу, вы должны написать:

IntegerVector test(IntegerVector labelling1) { 
    IntegerVector lvls = unique(labelling1); 
    return IntegerVector::create(lvls[1]); 
} 

Или вы могли бы также сделать, используя атрибуты Rcpp (который обрабатывает преобразование из int в IntegerVector автоматически для вас):

#include <Rcpp.h> 
using namespace Rcpp; 

// [[Rcpp::export]] 
int test(IntegerVector labelling1) { 
    IntegerVector lvls = unique(labelling1); 
    return lvls[1]; 
} 

/*** R 
test(1:5) 
*/ 

Answer 2

сахара Подменю ожидает IntegerVector в качестве индекса (http://gallery.rcpp.org/articles/subsetting/). Если вы хотите, чтобы имитировать функцию R в unique вам потребуются некоторые дополнительные изменения:

cppFunction("IntegerVector test(IntegerVector labelling1, int i) { 
       // get sorted unique values 
       IntegerVector lvls = sort_unique(labelling1); 
       // get unique values in order of occurence 
       IntegerVector lvls1 = lvls[match(lvls, labelling1) - 1]; 
       return(lvls1[IntegerVector::create(i - 1)]);}") 

test(c(5:1, 1L), 2) 
#[1] 4