将 std::vector 转换为 Rcpp 矩阵

Question

这是一个与 Rcpp 转换相关的 Q。我希望将 long std::vector 转换为 Rcpp 矩阵对象，但想知道是否有简单的转换格式。自然地，您可以遍历每个元素并填充一个空的 Rcpp 矩阵，但这似乎容易出错并且如果可能有更方便的方法可能是不必要的。

我问的原因是我想在一些现有的 C++ 代码中使用 OpenMP，但是将元素直接存储在 OpenMP 循环中的 Rcpp 矩阵对象中似乎效果不佳（而加载 std::vector对象并在 OpenMP 循环完成后转换为矩阵似乎是解决问题的一种不错的方法）。

任何帮助将不胜感激！

Answer 1

扩展德克的答案； R 矩阵实际上只是具有dim 属性集的向量

> x <- 1
> y <- as.matrix(x)
> .Internal( inspect(x) )
@7f81a6c80568 14 REALSXP g0c1 [MARK,NAM(2)] (len=1, tl=0) 1 ## data
> .Internal( inspect(y) )
@7f81a3ea86c8 14 REALSXP g0c1 [NAM(2),ATT] (len=1, tl=0) 1 ## data
ATTRIB:
  @7f81a3e68e08 02 LISTSXP g0c0 [] 
    TAG: @7f81a30013f8 01 SYMSXP g1c0 [MARK,LCK,gp=0x4000] "dim" (has value) 
    @7f81a3ea8668 13 INTSXP g0c1 [NAM(2)] (len=2, tl=0) 1,1

请注意x 和y 的“数据”组件只是REALSXPs，但是矩阵上有这个额外的dim 组件。使用它，我们可以轻松地将NumericVector 转换为Rcpp 中的矩阵。

（注意：在下面的示例中，我在属性中使用了SEXP，以便类型之间的转换是明确的）：

#include <Rcpp.h>
using namespace Rcpp;

// [[Rcpp::export]]
SEXP vec_to_mat(SEXP x_) {
  std::vector<double> x = as< std::vector<double> >(x_);
  /* ... do something with x ... */
  NumericVector output = wrap(x);
  output.attr("dim") = Dimension(x.size(), 1);
  return output;
}

/*** R
m <- c(1, 2, 3)
vec_to_mat(m)
*/

给我

> m <- c(1, 2, 3)

> vec_to_mat(m)
     [,1]
[1,]    1
[2,]    2
[3,]    3

因此，您可以使用 Dimension 类并将其分配给向量的 dim 属性，以在 Rcpp 中“手动”制作矩阵。

Answer 2

您选择了一个好方法。在 OpenMP 上下文中，你必须远离单线程 R。所以std::vector 很好。

是的，我们有从std::vector 到Rcpp::NumericVector 和Rcpp::NumericMatrix 的构造函数（这毕竟只是一个具有维度属性的向量）以及Rcoo:Integer* 变体，以及通常的@987654327 @ 和 wrap 转换器。

Rcpp 单元测试、Rcpp 示例，当然还有Rcpp Gallery 站点上应该有很多示例。

编辑：这是一个过于简单的例子：

 R> cppFunction('NumericVector phil(int n) { std::vector<double> x(n);
 +              return wrap(x); }') 
 R> phil(4)  
 [1] 0 0 0 0    
 R>