如何将 dplyr 与参数向量一起使用

Question

我正在尝试获取具有不同组的数据框，并找出哪些组共享一个共同值或 一组值。

例如，假设我有一些装有不同水果的水果篮，我想找出哪些篮子里有特定的水果。

这是我的水果篮：

fruit_baskets <- data.frame(basket=c(rep("blue",3),rep("red",3),rep("green",3)),
                        fruits=c("orange", "tomato", "pear", 
                                 "orange", "tomato", "grape",
                                 "strawberry", "tomato", "grape"))
# > print(fruit_baskets)
#   basket     fruits
# 1   blue     orange
# 2   blue     tomato
# 3   blue       pear
# 4    red     orange
# 5    red     tomato
# 6    red      grape
# 7  green strawberry
# 8  green     tomato
# 9  green      grape

还有我的函数来查找哪些篮子包含特定的水果：

find_similar_baskets_1 <- function(some_baskets, fruit) {
  some_baskets %>% 
    group_by(basket) %>% 
    summarise(xxx_1=max(ifelse(fruits==fruit, 1, 0))) %>% 
    filter(xxx_1 == 1) %>% 
    .[[1]] %>% unique %>% as.character
}

我可以使用这个功能找到装番茄的篮子。

> find_similar_baskets_1(fruit_baskets, "tomato")
[1] "blue"  "green" "red" 

同样，我可以创建一个函数来查找哪些篮子都有两个水果。

find_similar_baskets_2 <- function(some_baskets, fruit_1, fruit_2) {
  some_baskets %>% 
  group_by(basket) %>% 
  summarise(xxx_1=max(ifelse(fruits==fruit_1, 1, 0)),
            xxx_2=max(ifelse(fruits==fruit_2, 1, 0))) %>% 
  filter(xxx_1 == 1 & xxx_2 == 1) %>% 
  .[[1]] %>% unique %>% as.character
}

现在我可以使用这个函数来找出哪些篮子里有橙子和番茄（蓝色和红色的），哪些篮子里有番茄和葡萄（绿色和红色的）。

find_similar_baskets_2(fruit_baskets, "orange", "tomato")
#[1] "blue" "red" 

find_similar_baskets_2(fruit_baskets, "tomato", "grape")
#[1] "green" "red" 

在理想的世界中，我可以创建一个不依赖于我给它的值（即水果）数量的函数。

find_similar_groups <- function(df, group_column, value_column, value_vector)

我可以有一个任意大的值向量，而组列（在本例中为“篮子”）和值列（“水果”）不在函数中。

df = fruit_baskets
column = "fruits"
values = c("tomato")  or c("orange", "tomato")

Answer 1

这是另一种方式：

select_baskets_having_fruit <- function(bs,...){
    fs <- c(...)
    bs %>% 
        select_(basket, fruits) %>%
        group_by(basket) %>%
        filter(all(fs %in% fruits)) %>%
        .$basket %>% unique %>% as.character
}

还有一些例子：

fruit_baskets %>% select_baskets_having_fruit('tomato') 
# "blue"  "red"   "green"

fruit_baskets %>% select_baskets_having_fruit('orange','tomato') 
# "blue" "red"

fruit_baskets %>% select_baskets_having_fruit('tomato','grape') 
# "red"   "green"

---

泛化将此应用于其他分组和选择变量...

select_grp_with_x <- function(dat,...,g="basket",x="fruits"){
    xs <- c(...)
    dat %>% 
        select_(g,x) %>%
        group_by_(g) %>%
        filter(all(xs %in% .[[x]])) %>%
        .[[g]] %>% unique %>% as.character
}

fruit_baskets %>% select_grp_with_x('orange','tomato')

dplyr 在我看来不太适合制作泛化函数，因为您必须求助于.[[]] 之类的构造和select_ 之类的下划线函数。

例如，我更喜欢

library(magrittr)
select_grp_with_x2 <- function(dat,...,g,x)
    split(dat[[x]], dat[[g]]) %>% 
        sapply( function(z) all(c(...) %in% z) ) %>% 
        which %>% names

select_grp_with_x2(fruit_baskets,'orange','tomato',g='basket',x='fruits') # "blue" "red"

在了解基础 R 之后，我认为这比 dplyr 方式更容易阅读。但是，如果您正在设计一个提高效率的函数（快速运行而不消耗内存），您可能希望使用完全不同的方法。

Answer 2

这是一个答案，它以水果列表作为输入：

library(dplyr)
findbaskets <- function(df, fruitlist){
  x <- df %>% group_by(basket) %>%
              summarise(subset = all(fruitlist %in% fruits))
  x$basket[x$subset]
}
findbaskets(fruit_baskets, list("orange", "grape"))
[1] red

如果您想查看它是否包含all 列出的水果，请使用上面的内容，如果需要，请将all 更改为any。