Question

我正在尝试使用提供的维度列表将1D阵列映射到3D阵列。

以下是我的组件：

SEXP data; // my 1D array
// I can initialise new 3D vector in the following way:
NumericVector vector(Dimension(2, 2, 2);
// or the following:
NumericVector vector(data.begin(), data.end());

我没想到的是如何创建一个既有我的数据又有所需尺寸的NumericVector。

Answer 1

有一个较短的解决方案。您可以使用.attr重新整形数据。数据可以创建或作为输入提供 - 无关紧要。见下文：

library("Rcpp")

cppFunction(code='
NumericVector arrayC(NumericVector input, IntegerVector dim) { 
  input.attr("dim") = dim;
  return input;
}
')
x = 1:8
arrayC(x, c(2,2,2))
## , , 1
## 
##      [,1] [,2]
## [1,]    1    3
## [2,]    2    4
## 
## , , 2
## 
##      [,1] [,2]
## [1,]    5    7
## [2,]    6    8

Answer 2

这是可行的，但有点痛苦。我想对新构造函数或辅助函数的一个体面的（和测试的）贡献将是值得赞赏的。

与此同时，您可以执行以下示例所做的操作。但要注意行主要和主要等。另一个选项是RcppArmadillo，它有一个适当的'立方体'类型，将矩阵推广到3-d。

R> library(inline)
R> fx <- cxxfunction(signature(vs="numeric", ds="integer"), plugin="Rcpp", body='
+    Rcpp::NumericVector v(vs);            // get the data
+    Rcpp::Dimension d(ds);                // get the dim object
+    Rcpp::NumericVector r(d);             // create vec. with correct dims
+    std::copy(v.begin(), v.end(), r.begin());  // and copy
+    return Rcpp::List::create(v, d, r);
+ ')
R> fx(1:8, c(2,2,2))
[[1]]
[1] 1 2 3 4 5 6 7 8

[[2]]
[1] 2 2 2

[[3]]
, , 1

     [,1] [,2]
[1,]    1    3
[2,]    2    4

, , 2

     [,1] [,2]
[1,]    5    7
[2,]    6    8


R>

在Rcpp中构造3D数组

2 个答案: