Matriz RCPP: bucle sobre filas, una columna a la vez
Pregunta
Esta es la primera vez que intento RCPP y este problema muy simple me está dando problemas. Quiero usar bucles anidados para operar con valores individuales de una matriz, una columna a la vez. El script al que estoy apuntar se vería algo así:
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))
La salida deseada sería esta:
[,1] [,2] [,3] [,4]
[1,] 1 1 1 1
[2,] 2 2 2 2
[3,] 3 3 3 3
[4,] 4 4 4 4
El problema es obviamente en esta línea, donde no sé cómo referirme a elementos individuales de la matriz.
Am[j,i] = Am[j,i] + Am[j-1,i];
Disculpas si esta es una estúpida pregunta de novatos. ¡Cualquier pista sería apreciada!
No hay solución correcta
Licenciado bajo: CC-BY-SA con atribución
No afiliado a StackOverflow