R语言 如果不满足条件,则从列表中删除数据框

2lpgd968  于 2023-05-11  发布在  其他
关注(0)|答案(4)|浏览(146)

下面有一个列表对象。我想从列表中删除df2df3,因为它们没有“ID”列。如何解决这个问题?非常感谢。

my_list <- list(df1 = data.frame(ID = 1:5, Name = letters[1:5]),
                df2 = matrix(c(1, 2, 3, 4), ncol = 2),
                df3 = data.frame(Name = letters[6:10], Age = c(20, 25, 30, 35, 40)))

sapply(my_list, function(x) "ID" %in% colnames(x))

for (i in sequence(my_list)) {
  
  if (sapply(my_list, function(x) "ID" %in% colnames(x)) == FALSE) {
    DROP THE df2 and df3
  }
}
h5qlskok

h5qlskok1#

Filter() from base提取逻辑函数为真的列表中的元素。

Filter(\(df) "ID" %in% colnames(df), my_list)

purrr的等效选项是keep/discard

purrr::keep(my_list, ~ "ID" %in% colnames(.x))
purrr::discard(my_list, ~ !"ID" %in% colnames(.x))
输出
$df1
  ID Name
1  1    a
2  2    b
3  3    c
4  4    d
5  5    e
a7qyws3x

a7qyws3x2#

使用for循环

for(nm in names(my_list)) if(!"ID" %in% names(my_list[[nm]])) my_list[[nm]] <- NULL
f5emj3cl

f5emj3cl3#

您可以使用正常的逻辑子集:

my_list[sapply(my_list, function(x) "ID" %in% colnames(x))]
rqcrx0a6

rqcrx0a64#

另一种选择是使用布尔向量来子集列表。这里我们使用map_lgl

library(purrr)

my_list[map_lgl(my_list, ~ "ID" %in% colnames(.))]
$df1
  ID Name
1  1    a
2  2    b
3  3    c
4  4    d
5  5    e

相关问题