基于属性值的访问列表元素

Question

假设我有一个列表,其元素具有属性,如下所示:

my_list <- list()

my_list[[1]] <- 1:10
my_list[[2]] <- 11:20
my_list[[3]] <- 21:30

attr(my_list[[1]], "att1") <- "a"
attr(my_list[[2]], "att1") <- "b"
attr(my_list[[3]], "att1") <- "c"

attr(my_list[[1]], "att2") <- "1"
attr(my_list[[2]], "att2") <- "2"
attr(my_list[[3]], "att2") <- "3"

现在,假装这个列表长达数百个元素,我不知道列表中哪个元素具有我想要的属性.但我知道我想要元素,比如说att1 =="b"和att2 =="2"(但我不知道它恰好对应于列表元素2).

在R中是否有办法查找列表中的哪些元素具有特定的属性组合？

Answer 1

您可以使用以下内容过滤列表Filter:

Filter(function(x) attr(x, "att1") == "b" & attr(x, "att2") == "2", my_list)

如果您希望元素是唯一的并且想要选择它,请[[1]]在末尾添加.

就个人而言,我将数据放入表中:

library(data.table)
myDT = data.table(
  att1 = sapply(my_list, attr, "att1"), 
  att2 = sapply(my_list, attr, "att2"),
  data = my_list
)

#    att1 att2               data
# 1:    a    1       1,2,3,4,5,6,
# 2:    b    2 11,12,13,14,15,16,
# 3:    c    3 21,22,23,24,25,26,

然后你可以验证att1 + att2是否唯一地固定了一个元素

nrow(myDT) == uniqueN(myDT, by=c("att1", "att2"))
# [1] TRUE

并编写辅助函数进行子集化

setkey(myDT, att1, att2)
get_element = function(a1, a2) myDT[.(a1, a2), data[[1]]]

get_element("b", "2")
#  [1] 11 12 13 14 15 16 17 18 19 20
# attr(,"att1")
# [1] "b"
# attr(,"att2")
# [1] "2"

您可能还需要查看purrr和broom包,它们为具有列表列的表提供不同的"tidyverse"语法.