如何基于过滤器从数据框创建命名列表

Question

我的问题是如何简化下面的代码。

我有一个包含两个（或更多）列的数据框，其中我对列中的所有值感兴趣，并且＆＃39; a＆＃39;列＆＃39; b＆＃39;有价值x。 由于x可能是一个或多个值，我想在所有可能的过滤器中构建一个命名列表。

所以，我是R的新手，我解决了这个问题，但我认为这不是最好的解决方案。

a <- c('a', 'b','c','d','e','f','g','h','i')
b <- c('aa', 'bb','cc')
ds <- data.frame(a, b)

func <- function(ds, target_col, query_col, value){ 
    return (unlist(list(unique(as.vector(ds[ds[query_col] == value,][target_col][,1])))))
}

named_list <- list()
for (i in b){
    named_list[[i]] <- func(ds, 'a', 'b', i)
}

Answer 1

func可以简化;以下是等效的，假设target_col总是长度为1：

func <- function(ds, target_col, query_col, value){ 
    return (unique(as.vector(ds[ds[query_col] == value, target_col])))
}

您可以使用lapply功能

创建命名列表

named_list2 <- lapply(setNames(nm = b), func, ds = ds, target_col = "a", query_col = "b")

其中setNames创建一个命名向量，其名称与值相同; lapply使用此向量中的名称输出命名列表。