Rcpp中的lengths（）糖函数？

Question

在基数R中，lengths（）函数计算列表中每个元素的长度。例如，

> mylist = list(v1 = c(1,2,3), v2 = c(1,2), v3 = c(1,2,3,4))
> mylist
$`v1`
[1] 1 2 3

$v2
[1] 1 2

$v3
[1] 1 2 3 4

> lengths(mylist)
v1 v2 v3 
 3  2  4 

Rcpp中是否有类似的功能（或从Armadillo获得）？我知道可以编写一个基于Rcpp的apply（）函数，但是我想在这一步优化速度。谢谢！

Answer 1

我不知道等效的糖功能，因此非常想知道是否也存在糖功能。我知道您说您可以编写自己的东西，但是我将通过使用的这两个功能将我的帽子戴上。一个是递归的，因此它确定每个子列表元素的大小，另一个等效于lengths()（非递归）。

library(Rcpp)

cppFunction(

  code = 'Rcpp::List rcpp_lengths_recursive( const Rcpp::List& lst ) {
  std::size_t n = lst.size();
  Rcpp::List res( n ); 
  std::size_t i;

  for( i = 0; i < n; i++ ) {
    switch( TYPEOF( lst[i] ) ) {
      case VECSXP: {
        // might need an `Rf_inherits( lst[i], "data.frame" )` check, 
        // depending on your use-case
        res[ i ] = rcpp_lengths_recursive( lst[i] );
        break;
      }
      default: {
        int n_elements = Rf_length( lst[i] );
        res[i] = n_elements;
      }
    }
  }
  return res;
}'

)

cppFunction(

  code = 'Rcpp::IntegerVector rcpp_lengths( const Rcpp::List& lst ) {
    std::size_t n = lst.size();
    Rcpp::IntegerVector res( n );
    std::size_t i;

    for( i = 0; i < n; i++ ) {
      res[i] = Rf_length( lst[i] );
    }
    return res;
  }'

)

mylist <- list(v1 = c(1,2,3), v2 = c(1,2), v3 = list(list(c(1,2,3,4)), 1:2))

rcpp_lengths( mylist )
# [1] 3 2 2

rcpp_lengths_recursive( mylist )
# [[1]]
# [1] 3
# 
# [[2]]
# [1] 2
# 
# [[3]]
# [[3]][[1]]
# [[3]][[1]][[1]]
# [1] 4
# 
# 
# [[3]][[2]]
# [1] 2