r - 迭代向量的笛卡尔积

Question

我有以下嵌套循环:

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

我想将其展平，因此我想到了构建两个向量 xs和 ys的笛卡尔积，并对结果进行迭代。在Python中，这很简单:

for xy in product(xs, ys):
    # x, y = xy[0], xy[1]

但是在R中，我发现的最简单的等效项看起来令人生畏:

xys <- expand.grid(xs, ys)
for (i in 1 : nrow(xys)) {
    xy <- as.vector(xys[i, ])
    # x <- xy[1], y <- xy[2]
}

当然必须有更好的方法，不是吗？ (为澄清起见，我不想遍历索引...我认为必须有一种直接遍历产品中元组的方法。)

Answer 1

您可以使用apply函数将一个函数应用于数据框的每一行。只需将"your function"替换为您的实际功能即可。

# example data
xs <- rnorm(10)
ys <- rnorm(10)    

apply(expand.grid(xs, ys), 1, FUN = function(x) {"your function"})

这是一个非常基本的例子。此处，计算一行中两个值的总和:

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

这是一个使用命名参数( xs， ys)而不是索引( 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)))