Question

我正在编写一个聚合数据框的函数，它通常需要适用于各种各样的数据集。此功能的一个步骤是dplyr的filter功能，用于仅从数据中选择与手头任务相关的广告活动类型。由于我需要灵活的功能，我想要ad_campaign_types作为输入，但这会使过滤变得多毛，因为：

aggregate_data <- function(ad_campaign_types) {
  raw_data %>%
    filter(ad_campaign_type == ad_campaign_types) -> agg_data
  agg_data
}
new_data <- aggregate_data(ad_campaign_types = c("campaign_A", "campaign_B", "campaign_C"))

我认为上面的方法可行，但是当它运行时，奇怪的是它只返回过滤数据集的一小部分。有更好的方法吗？

可替换代码的另一个小例子：

ad_types <- c("a", "a", "a", "b", "b", "c", "c", "c", "d", "d")
revenue <- c(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
data <- as.data.frame(cbind(ad_types, revenue))

# Now, filtering to select only ad types "a", "b", and "d",
# which should leave us with only 7 values
new_data <- filter(data, ad_types == c("a", "b", "d"))
nrow(new_data)
[1] 3

Answer 1

对于多个条件，请使用%in%函数：

filter(data, ad_types %in% c("a", "b", "d"))

你也可以使用“不在”标准：

filter(data, !(ad_types %in% c("a", "b", "d")))

但请注意%in%的行为与==略有不同：

> c(2, NA) == 2
[1] TRUE   NA
> c(2, NA) %in% 2
[1]  TRUE FALSE

有些人发现其中一个比其他人更直观，但你必须记住差异。

至于使用多个不同的标准，只需使用标准链和/或语句：

filter(mtcars, cyl > 2 & wt < 2.5 & gear == 4)

Answer 2

Tim对于过滤数据帧是正确的。但是，如果您想使用dplyr创建一个函数，则需要按照此网页上的说明进行操作：https://rpubs.com/hadley/dplyr-programming。

我建议的代码。

library(tidyverse)
ad_types <- c("a", "a", "a", "b", "b", "c", "c", "c", "d", "d")
revenue <- c(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
df <- data_frame(ad_types = as.factor(ad_types), revenue = revenue)

aggregate_data <- function(df, ad_types, my_list) {
      ad_types = enquo(ad_types) # Make ad_types a quosure
      df %>%
          filter(UQ(ad_types) %in% my_list) # Unquosure
}

new_data <- aggregate_data(df = df, ad_types = ad_types, 
                           my_list = c("a", "b", "c"))

这应该有用！

如何通过多个灵活的条件筛选列

2 个答案: