Iterate over cartesian product of vectors

后端 未结 2 730
感情败类
感情败类 2021-01-02 22:18

I have the following nested loop:

for (x in xs) {
    for (y in ys) {
        # Do something with x and y
    }
}

Which I’d like to flatten

2条回答
  •  一个人的身影
    2021-01-02 22:53

    You can use the apply function to apply a function to each row of your data frame. Just replace "your function" with your actual function.

    # example data
    xs <- rnorm(10)
    ys <- rnorm(10)    
    
    apply(expand.grid(xs, ys), 1, FUN = function(x) {"your function"})
    

    This is a very basic example. Here, the sum of both values in a row is calculated:

    apply(expand.grid(xs, ys), 1, FUN = function(x) {x[1] + x[2]})
    

    Here is a variant that uses named arguments (xs, ys) instead of indices (x[1], x[2]):

    myfun <- function(xs, ys) xs + ys
    arguments <- expand.grid(xs = rnorm(10), ys = rnorm(10))
    apply(arguments, 1, function(x)do.call(myfun, as.list(x)))
    

提交回复
热议问题