我如何将矩阵列表从RCPP返回到R

Question

我在RCPP中具有这样的功能：它创建了类型std :: list的矩阵列表，并打算将矩阵列表返回r。p>我在这里附上一个简化的示例：

#include <RcppArmadillo.h>
// [[Rcpp::depends(RcppArmadillo)]]
// [[Rcpp::export]]
Rcpp::List splitImagesRcpp(arma::mat x) 
{
   std::list<arma::mat> listOfRelevantImages;
   int relevantSampleSize = x.n_rows;
   for(int k = 0; k < relevantSampleSize; ++k)
   {
      listOfRelevantImages.push_back(x.row(k));
   }
   return wrap(listOfRelevantImages);
}

这里的问题是，我想返回r矩阵列表，但是我得到了向量列表。我一直在尝试很多东西并查看文档，但是我似乎找不到解决方案。看起来包裹正在做工作，但它也将我的矩阵递归包装在列表中。

我得到这样的东西：

> str(testingMatrix)
List of 200
 $ : num [1:400] 1 1 1 1 1 1 1 1 1 1 ...
 $ : num [1:400] 1 1 1 1 1 1 1 1 1 1 ...

，但我想得到这样的东西：

> str(testingMatrix)
List of 200
 $ : num [1:40, 1:10] 1 1 1 1 1 1 1 1 1 1 ...
 $ : num [1:40, 1:10] 1 1 1 1 1 1 1 1 1 1 ...

我想从RCPP而不是在R中进行此操作。那是因为我希望能够与纯R编程的功能互换以测量加速。

任何帮助都将不胜感激！

Answer 1

使用具有必须转换为转换的arma::field类，并通过 r 和 c 。。。

这是一些示例代码，涉及一个人如何与现场类一起工作，因为您上面的示例不可再现...

#include <RcppArmadillo.h>
// [[Rcpp::depends(RcppArmadillo)]]
// [[Rcpp::export]]
arma::field<arma::mat> splitImagesRcpp(arma::mat x) {
    // Sample size
    int relevantSampleSize = x.n_rows;
    // Create a field class with a pre-set amount of elements
    arma::field<arma::mat> listOfRelevantImages(relevantSampleSize);
    for(int k = 0; k < relevantSampleSize; ++k)
    {
        listOfRelevantImages(k) = x.row(k);
    }

    return listOfRelevantImages;
}

示例：

set.seed(1572)
(x = matrix(runif(25), 5, 5))
#           [,1]        [,2]      [,3]      [,4]       [,5]
# [1,] 0.2984725 0.679958392 0.5636401 0.9681282 0.25082559
# [2,] 0.3657812 0.157172256 0.6101798 0.5743112 0.62983179
# [3,] 0.6079879 0.419813382 0.5165553 0.3922179 0.64542093
# [4,] 0.4080833 0.888144280 0.5891880 0.6170115 0.13076836
# [5,] 0.8992992 0.002045309 0.3876262 0.9850514 0.03276458
(y = splitImagesRcpp(x))
#      [,1]     
# [1,] Numeric,5
# [2,] Numeric,5
# [3,] Numeric,5
# [4,] Numeric,5
# [5,] Numeric,5
y[[1]]
#           [,1]      [,2]      [,3]      [,4]      [,5]
# [1,] 0.2984725 0.6799584 0.5636401 0.9681282 0.2508256