2

我正在尝试编写一个函数，该函数需要从结果数据框中排除用户传递的变量。我也借此机会进一步了解了新的 dplyr 语法。

该函数的作用类似于数据帧的交叉连接。我想用它作为跨函数参数复制数据的干净方式。

该功能的工作原理如下：

crossjoin_df <- function(df1, df2, temp_col = ".k") {
  df1 <- df1 %>% 
    mutate(!!temp_col :=  1)

  df2 <- df2 %>% 
    mutate(!!temp_col :=  1)

  out <- left_join(df1, df2, by = temp_col)

  # I'm trying to replace the next line
  out[,!names(out)==temp_col] 
} 

params <- data.frame(k = c(11,10),
                 n = c(27,26))

data <- data.frame(a = 1:3,
               b = 4:6)

crossjoin_df(params, data) # 6 row data set

我想看看是否可以用管道选择语句替换最后一个语句。但是，否定似乎不起作用。

我能够得到类似的东西：

out %>% select(!!temp_col)

工作，但这显然只选择.k. 我无法得到类似的东西：

out %>% select(-!!temp_col)

去工作。

4

3 回答 3

2

无论您是否想继续使用字符串，您都需要 rlang，它是 dplyr 的后端包，它可以启用 tidy eval，在这种情况下，您需要sym将字符串转换为 quosure：

library(dplyr)

params <- data.frame(k = c(11,10),
                     n = c(27,26))

data <- data.frame(a = 1:3,
                   b = 4:6)

crossjoin_df <- function(df1, df2, temp_col = ".k") {
    df1 <- df1 %>% mutate(!!temp_col :=  1)

    df2 <- df2 %>% mutate(!!temp_col :=  1)

    left_join(df1, df2, by = temp_col) %>% 
        select(-!!rlang::sym(temp_col))
}

crossjoin_df(params, data)
#>    k  n a b
#> 1 11 27 1 4
#> 2 11 27 2 5
#> 3 11 27 3 6
#> 4 10 26 1 4
#> 5 10 26 2 5
#> 6 10 26 3 6

...或切换到 full tidy eval，在这种情况下，您需要quo_name将 quosure 转换为名称：

crossjoin_df <- function(df1, df2, temp_col = .k) {
    temp_col <- enquo(temp_col)

    df1 <- df1 %>% mutate(!!rlang::quo_name(temp_col) :=  1)

    df2 <- df2 %>% mutate(!!rlang::quo_name(temp_col) :=  1)

    left_join(df1, df2, by = rlang::quo_name(temp_col)) %>% 
        select(-!!temp_col)
}

crossjoin_df(params, data)
#>    k  n a b
#> 1 11 27 1 4
#> 2 11 27 2 5
#> 3 11 27 3 6
#> 4 10 26 1 4
#> 5 10 26 2 5
#> 6 10 26 3 6

或者，只需使用tidyr::crossing：

tidyr::crossing(params, data)
#>    k  n a b
#> 1 11 27 1 4
#> 2 11 27 2 5
#> 3 11 27 3 6
#> 4 10 26 1 4
#> 5 10 26 2 5
#> 6 10 26 3 6
于 2017-11-24T18:06:38.263 回答
1

您可以使用one_of，然后使用 否定选择-：

out %>% select(-one_of(temp_col))

crossjoin_df <- function(df1, df2, temp_col = ".k") {
  # `$`(df1, temp_col) <- 1
  df1 <- df1 %>% 
    mutate(!!temp_col :=  1)

  # `$`(df2, temp_col) <- 1
  df2 <- df2 %>% 
    mutate(!!temp_col :=  1)

  left_join(df1, df2, by = temp_col) %>% select(-one_of(temp_col))

} 

params <- data.frame(k = c(11,10),
                 n = c(27,26))

data <- data.frame(a = 1:3,
               b = 4:6)

crossjoin_df(params, data)

#   k  n a b
#1 11 27 1 4
#2 11 27 2 5
#3 11 27 3 6
#4 10 26 1 4
#5 10 26 2 5
#6 10 26 3 6
于 2017-11-24T17:30:06.873 回答
0

这也应该有效：

out %>% select_(paste0("-",temp_col))
于 2017-11-24T17:40:36.310 回答