简体   繁体   中英

loop though a matrix in R

I try to loop trough a matrix but cant find a easy and elegant way instead of writing many (>10) equations... Can anyone help me please?

My Matrix looks like this:

在此处输入图像描述

and I want to calculate the following:

(0 * 0 * 4/24) + (0 * 1 * 6/24) + (0 * 2 * 3/24) + (1 * 0 * 3/24) + (1 * 1 * 4/24) + (1 * 2 * 4/24)

instead of using

__ btw: my code for the matrix

vals<- c(4/24, 6/24, 3/24, 3/24, 4/24, 4/24)
x <- c(0,1)
y <- c(0,1,2)
df <- matrix(vals, byrow = TRUE, nrow = 2, ncol = 3,
             dimnames = list(x,y))

instead of calculation each step manually, I think there should be a for-loop method, but cant figure it out..

A possible solution:

c(x %*% df %*% y) 

#> [1] 0.5

Another possible solution, based on outer :

sum(outer(x, y, Vectorize(\(x,y) x*y*df[x+1,y+1])))

#> [1] 0.5
x <- c(0, 1)
y <- c(0, 1, 2)
vals<- c(4/24, 6/24, 3/24, 3/24, 4/24, 4/24)
mat <- matrix(vals, byrow = TRUE, nrow = 2, ncol = 3,
              dimnames = list(x,y))  ## not a data frame; don't call it "df"

There is even a better way than a for loop:

sum(tcrossprod(x, y) * mat)
#[1] 0.5
sum((x %o% y) * df)

Explanation:

x %o% y gets the outer product of vectors x and y which is:

#>      [,1] [,2] [,3]
#> [1,]    0    0    0
#> [2,]    0    1    2

Since that has the same dimensions as df , you can multiply the corresponding elements and get the sum: sum((x %o% y) * df)

If you are new to R (as I am), here is the loop approach.

result = 0
for (i in 1:length(x)) {
  for (j in 1:length(y)) {
    result = result + x[i] * y[j] * df[i, j]
  }
}
result

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM