программирование dplyr с использованием mutate - PullRequest
0 голосов
/ 07 июня 2018

Я борюсь за программирование с помощью dplyr с помощью mutate.

Ниже приведен воспроизводимый пример.

library(tidyverse)
test <- tibble(a=1:10, b=1:10)

mute_function <- function(var, table) {

  quo_var <- enquo(var)

  sub_tb <-
    table %>%
    dplyr::select(!!quo_var, b) %>%
    mutate(new_var=!!quo_var)

  return(sub_tb)
}

При применении функции

mute_function(var = "a", table = test)

получаю:

# A tibble: 10 x 3
       a     b new_var
   <int> <int> <chr>  
 1     1     1 a      
 2     2     2 a      
 3     3     3 a      
 4     4     4 a      
 5     5     5 a      
 6     6     6 a      
 7     7     7 a      
 8     8     8 a      
 9     9     9 a      
10    10    10 a  

Я хочу получить новую переменную, объединяющую значения из a и других переменных.В вышеприведенном случае я ожидал new_var идентичного a.

Очевидно, что следующее не работает:

mute_function <- function(var, table) {

  quo_var <- enquo(var)

  sub_tb <-
    table %>%
    dplyr::select(!!quo_var, b) %>%
    mutate(new_var=!!quo_var/b)

  return(sub_tb)
}

 mute_function(var = "a", table = test)

Я не понимаю, почему я былвнимательно читайте dplyr виньетка для программирования.

Заранее спасибо за помощь

1 Ответ

0 голосов
/ 07 июня 2018

Если мы передаем символьную строку, используйте sym для преобразования в symbol вместо выражения

mute_function <- function(var, table) {

  quo_var <- rlang::sym(var)

  sub_tb <-
    table %>%
    dplyr::select(!!quo_var, b) %>%
    mutate(new_var=!!quo_var/b)

  return(sub_tb)
}

out <- mute_function(var = "a", table = test)
out
# A tibble: 10 x 3
#       a     b new_var
#   <int> <int>   <dbl>
# 1     1     1       1
# 2     2     2       1
# 3     3     3       1
# 4     4     4       1
# 5     5     5       1
# 6     6     6       1
# 7     7     7       1
# 8     8     8       1
# 9     9     9       1
#10    10    10       1

Но, если целью является передача аргументов без кавычек, функция OP должнаработать правильно

mute_function(var = a, table = test)

Однако мы можем передавать строки как в кавычках, так и без кавычек с

mute_function <- function(var, table) {

  quo_var <- rlang::parse_expr(quo_name(enquo(var)))

  sub_tb <-
    table %>%
    dplyr::select(!!quo_var, b) %>%
    mutate(new_var=!!quo_var/b)

  return(sub_tb)
}

out1 <- mute_function(var = "a", table = test)
out2 <- mute_function(var = a, table = test)
identical(out1, out2)
#[1] TRUE
identical(out, out1)
#[1] TRUE
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...