假设我有一个列表x
如下:
library(purrr)
set.seed(4152)
x <- rerun(5, a = sample(c("A","B","C","D"),1), b = c(1,2,3,4,5))
x
[[1]]
[[1]]$a
[1] "B"
[[1]]$b
[1] 1 2 3 4 5
[[2]]
[[2]]$a
[1] "B"
[[2]]$b
[1] 1 2 3 4 5
[[3]]
[[3]]$a
[1] "C"
[[3]]$b
[1] 1 2 3 4 5
[[4]]
[[4]]$a
[1] "C"
[[4]]$b
[1] 1 2 3 4 5
[[5]]
[[5]]$a
[1] "A"
[[5]]$b
[1] 1 2 3 4 5
我想keep
x
列表$a %in% c("A","C")
中的所有元素,所以我试过:
x_sub <- x %>%
keep(., "a" %in% c("A","C"))
Error: length(.p) == length(.x) is not TRUE
基于我对keep
函数的理解,我认为这是一种自然的编码方式。为什么报告错误?任何建议表示赞赏。
您有正确的想法,但需要利用as_mapper()
创建一个谓词函数,该函数将访问列表中您想要的值。我们不会在c(“A”,“C”)中说“a”,而是会说x$a in c("A","C")
差别很小,但这样才有效。
library(tidyverse)
set.seed(4152)
x <- rerun(5, a = sample(c("A","B","C","D"),1), b = c(1,2,3,4,5))
x_sub <- x %>%
keep(., as_mapper(~.x$a %in% c("A","C")))
x_sub
#> [[1]]
#> [[1]]$a
#> [1] "C"
#>
#> [[1]]$b
#> [1] 1 2 3 4 5
#>
#>
#> [[2]]
#> [[2]]$a
#> [1] "C"
#>
#> [[2]]$b
#> [1] 1 2 3 4 5
#>
#>
#> [[3]]
#> [[3]]$a
#> [1] "A"
#>
#> [[3]]$b
#> [1] 1 2 3 4 5
由reprex package创建于2019-05-01(v0.2.1)
这是base R
中的一个选项
Filter(function(y) y$a %in% c('A', 'C'), x)
#[[1]]
#[[1]]$a
#[1] "C"
#[[1]]$b
#[1] 1 2 3 4 5
#[[2]]
#[[2]]$a
#[1] "C"
#[[2]]$b
#[1] 1 2 3 4 5
#[[3]]
#[[3]]$a
#[1] "A"
#[[3]]$b
#[1] 1 2 3 4 5
为了完整性,这里有一个purrr
,plyr
替代方案:
compact(plyr::llply(x,function(x) x[x$a %in% c("A","C")]))
[[1]]$a
[1] "C"
[[1]]$b
[1] 1 2 3 4 5
[[2]]
[[2]]$a
[1] "C"
[[2]]$b
[1] 1 2 3 4 5
[[3]]
[[3]]$a
[1] "A"
[[3]]$b
[1] 1 2 3 4 5