r - 在 Rcpp 中构造 3D 数组

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

这是可行的，但有点痛苦。我想对新的构造函数或辅助函数的体面（和测试）贡献将不胜感激。

同时，您可以执行以下示例的操作。但要小心 row-major 和 col-major 等。另一个选择是RcppArmadillo，它有一个适当的 'Cube' 类型将矩阵泛化到 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>