[R] Matrix indexing in a loop
Pontarelli, Brett
bpontar at amazon.com
Sat Feb 18 00:04:18 CET 2006
Do you have to use a loop? The following function should do what you want for the 1st order:
rook = function(Y) {
rsub = function(Z) {
X = matrix(0,nrow(Z),ncol(Z));
X[1:(N-1),1:M] = X[1:(N-1),1:M] + Z[2:N,1:M];
X[2:N,1:M] = X[2:N,1:M] + Z[1:(N-1),1:M];
X[1:N,1:(M-1)] = X[1:N,1:(M-1)] + Z[1:N,2:M];
X[1:N,2:M] = X[1:N,2:M] + Z[1:N,1:(M-1)];
return(X);
}
return(rsub(Y)/rsub(matrix(1,nrow(Y),ncol(Y))));
}
I'm not sure I understand how the higher orders work. For example, an interior element for the 1st order is always divided by 4. Is an interior element for a 3rd order divided by 4 or 8 or something else? Also, how are you implementing your 3D matrices?
--Brett
-----Original Message-----
From: r-help-bounces at stat.math.ethz.ch [mailto:r-help-bounces at stat.math.ethz.ch] On Behalf Of Mills, Jason
Sent: Friday, February 17, 2006 1:36 PM
To: r-help at stat.math.ethz.ch
Subject: [R] Matrix indexing in a loop
How do you specify matrix location a[i,j] (or a[i-1,j], etc.) in a "for"
loop?
I am looking for a flexible method of indexing neighbors over a series of lags (1,2,3...) and I may wish to extend this method to 3D arrays.
Example:
Data matrix
> fun
[,1] [,2] [,3]
[1,] 1 5 9
[2,] 2 6 10
[3,] 3 7 11
[4,] 4 8 12
For each element a[i,j] in "fun", sum the 1st order (Rook's) neighbors:
a[i-1,j]
a[i+1,j]
a[i,j-1]
a[i,j+1]
Then divide by the number of elements included as neighbors-- this number depends on the location of a[i,j] in the matrix.
Insert the product of the neighbor calculation for each a[i,j] into the corresponding position b[i,j] in an empty matrix with the same dimensions as "fun".
For example, element [2,2] in "fun" should yield element [2,2] in a new matrix equal to 24/4=6. Of course, element [1,1] in the new matrix should be the product of only two numbers.
Thanks
J. Mills
[[alternative HTML version deleted]]
______________________________________________
R-help at stat.math.ethz.ch mailing list
https://stat.ethz.ch/mailman/listinfo/r-help
PLEASE do read the posting guide! http://www.R-project.org/posting-guide.html
More information about the R-help
mailing list