从R中的列表中进行选择性消除

时间:2019-01-29 01:14:02

标签: r list function lapply

我想知道如何从列表{中的第二个变量(在本例中为x,即x[[2]])上删除第二个变量0:90元素 {1}}对应的xy

0

P.S。我的目标是可以使用x = list(0:5, 0:90) # from the second variable on, in this list, eliminate elements whose # corresponding `y` is `0` ? y = lapply(list(dbinom(x[[1]], 5, .9), dpois(x[[2]], 50)), round, digits = 4) 来访问更大的列表。

1 个答案:

答案 0 :(得分:1)

在这种情况下,您可以

x[[2]][y[[2]] != 0]

获得您的预期输出。

但是,如上所述,您有一个较大的列表,并且希望对每个列表都执行此操作。在这种情况下,我们可以使用mapply

mapply(function(p, q) p[q != 0], x[2:length(x)], y[2:length(y)], SIMPLIFY = FALSE)

或者,如果我们想使用lapply,我们可以

lapply(2:length(x), function(i) x[[i]][y[[i]] != 0])

如果我们想保持第一个元素不变就可以了

c(list(x[[1]]), lapply(2:length(x), function(i) x[[i]][y[[i]] != 0]))

编辑

要保持顺序,我们可以根据x重新安排ysmallest_max

get_new_list <- function(x, y) {
   smallest_max <- which.min(sapply(x, max))
   new_x <- c(x[smallest_max], x[-smallest_max])
   new_y <- c(y[smallest_max], y[-smallest_max])
  c(new_x[1], lapply(2:length(new_x), function(i) new_x[[i]][new_y[[i]] != 0]))
}

x = list(0:5, 0:40)
y = lapply(list(dbinom(x[[1]], 5, .9), dpois(x[[2]], 50)), round, digits = 4)

get_new_list(x, y)
#[[1]]
#[1] 0 1 2 3 4 5

#[[2]]
#[1] 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40

x = list(0:40, 0:5)
y = lapply(list(dpois(x[[1]], 50), dbinom(x[[2]], 5, .9)), round, digits = 4)

get_new_list(x, y)
#[[1]]
#[1] 0 1 2 3 4 5

#[[2]]
#[1] 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40