Rcpp 矩阵：循环行，一次一列

Question

这是我第一次尝试 Rcpp，这个非常简单的问题给我带来了麻烦。我想使用嵌套的 for 循环对矩阵的各个值进行操作，一次一列。我的目标脚本看起来像这样：

src <- '
    Rcpp::NumericMatrix Am(A);
    int nrows = Am.nrow();
    int ncolumns = Am.ncol();
    for (int i = 0; i < ncolumns; i++){
        for (int j = 1; j < nrows; j++){
            Am[j,i] = Am[j,i] + Am[j-1,i];
        }
    }
    return Am;
'
fun <- cxxfunction(signature(A = "numeric"), body = src, plugin="Rcpp")
fun(matrix(1,4,4))

所需的输出将是这样的：

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

问题显然出在这一行，我不知道如何引用矩阵的各个元素。

Am[j,i] = Am[j,i] + Am[j-1,i];

如果这是一个愚蠢的新手问题，我深表歉意。任何提示将不胜感激！

Answer 1

您不能在单个[ ] 表达式中使用多个索引。这是我所知道的 C++ 矩阵类系统或库无法克服的 C 语言限制。因此请使用 ( ) 来代替。

修复这个问题以及您实际上没有将 src 传递给 cxxfunction() 的错误，我们得到：

R> src <- '
+     Rcpp::NumericMatrix Am(A);
+     int nrows = Am.nrow();
+     int ncolumns = Am.ncol();
+     for (int i = 0; i < ncolumns; i++) {
+         for (int j = 1; j < nrows; j++) {
+             Am(j,i) = Am(j,i) + Am(j-1,i);
+         }
+     }
+     return Am;
+ '
R> fun <- cxxfunction(signature(A = "numeric"), body = src, plugin="Rcpp")
R> fun(matrix(1,4,4))
     [,1] [,2] [,3] [,4]
[1,]    1    1    1    1
[2,]    2    2    2    2
[3,]    3    3    3    3
[4,]    4    4    4    4
R> 

最后，请注意，Rcpp Sugar 有用于处理整行或的示例每次列一次，请参阅邮件列表档案和小插图。

编辑： 明确地说，这里仅使用一个循环和 Rcpp 糖的按列索引：

R> src <- '
+     Rcpp::NumericMatrix Am(A);
+     int nrows = Am.nrow();
+     for (int j = 1; j < nrows; j++) {
+         Am(j,_) = Am(j,_) + Am(j-1,_);
+     }
+     return Am;
+ '
R> fun <- cxxfunction(signature(A = "numeric"), body = src, plugin="Rcpp")
R> fun(matrix(1,4,4))
     [,1] [,2] [,3] [,4]
[1,]    1    1    1    1
[2,]    2    2    2    2
[3,]    3    3    3    3
[4,]    4    4    4    4
R>