我有以下嵌套循环:
- 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]
- }
解决方法
您可以使用apply函数将函数应用于数据框的每一行.只需将“您的功能”替换为您的实际功能即可.
- # example data
- xs <- rnorm(10)
- ys <- rnorm(10)
- apply(expand.grid(xs,ys),1,FUN = function(x) {"your function"})
这是一个非常基本的例子.在这里,计算一行中两个值的总和:
- apply(expand.grid(xs,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,function(x)do.call(myfun,as.list(x)))