如何对列表中的项目进行子集化

Question

考虑。

li <- list(c(1,1,1), c(2,1,1), c(5,1,0))
## [[1]]
## [1] 1 1 1                <-- 0 modulo 3
## 
## [[2]]
## [1] 2 1 1
## 
## [[3]]
## [1] 5 1 0                <--- 0 modulo 3

我想保留列表中的所有项目，其总和模 3 等于零（或关于列表中项目的一些其他逻辑表达式）。

代码。

  new <- list()
  idx <- 1
  for (i in seq_along(li) ) {
    nxt <- li[[i]]
    if ((sum(nxt) %% 3) == 0) {
      new[idx] <- list(nxt)
      idx <- idx + 1
    } 
  }
## new
## [[1]]
## [1] 1 1 1
##
## [[2]]
## [1] 5 1 0

但是。我想要一个矢量化的解决方案。 我试过了。

new <- subset(li, lapply(li, (sum %% 3) == 0 ))

但无济于事

编辑 另请参阅来自 dplyr 的子集和过滤器之间的区别以获取更多详细信息。

Answer 1

在基地R：

Filter(\(x)!sum(x)%%3, li)
[[1]]
[1] 1 1 1

[[2]]
[1] 5 1 0

整洁的宇宙

purrr::keep(li, ~!sum(.x)%%3)
[[1]]
[1] 1 1 1

[[2]]
[1] 5 1 0

purrr::discard(li, ~!!sum(.x)%%3) # !0
[[1]]
[1] 1 1 1

[[2]]
[1] 5 1 0