r mapply vs“bad lapply”

问题描述 投票:6回答:1

我想我在这里错过了一些简单的东西:我有一个data.frames列表,以及一个要选择的行号列表。像这样的东西:

a <- data.frame(q = c(1,0,0,0), 
                w = c(1,1,0,0),
                e = c(1,1,1,0),
                r = c(1,1,1,1))
b <- a + 1
c <- a + 2
d <- a + 3

data <- list(a = a, b = b, c = c, d = d)

ind_a <- c(1, 2) 
ind_b <- c(1, 3)
ind_c <- c(1, 4)
ind_d <- c(2, 4)

train <- list(ind_a, ind_b, ind_c, ind_d) 

现在,我想选择行,我认为正确的形式可能是

test1 <- mapply(function(x,y) x[y, ], data, train)

但我能使它发挥作用的唯一方法是

test2 <- lapply(1:4, function(x) data[[x]][train[[x]], ])

这对我来说似乎是假的for-loop ...

哪里我错了?

r vectorization lapply mapply
1个回答
4
投票

使用mapply,默认选项是SIMPLIFY = TRUE,当尺寸相同时,它将其简化为数组。如果我们将它改为FALSE,输出将是list

mapply(function(x,y) x[y, ], data, train, SIMPLIFY = FALSE)

或者使用Map包装

Map(function(x, y) x[y, ], data, train)
© www.soinside.com 2019 - 2024. All rights reserved.