Rcpp matrix: цикл по строкам, по одному столбцу за раз

Это моя первая попытка Rcpp, и эта очень простая проблема дает мне проблемы. Я хочу использовать вложенные для циклов для работы с отдельными значениями матрицы, по одному столбцу за раз. script Я стараюсь выглядеть примерно так:

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];

Извините, если это глупый вопрос новичков. Любой намек будет оценен!

Ответы

Ответ 1

Вы не можете использовать несколько индексов в одном выражении [ ]. Это ограничение на C-язык, что нет C32-матричная система или библиотека, которую я знаю о победах. Поэтому используйте ( ).

Фиксируя это и ошибку, которую вы фактически не пропустили 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 есть примеры для работы во всей строке или столбце за раз, см. архивы списков рассылки и виньетку.

Изменить: Просто для того, чтобы быть явным, это то же самое, используя только один цикл и индексирование столбца сахара 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>