Несколько имен столбцов к кво в dplyr NSE - PullRequest
Новая
       6

Несколько имен столбцов к кво в dplyr NSE

0 голосов
/ 08 мая 2018

Я пишу функции для автоматизации рабочего процесса для анализа большого количества демографических данных.Я могу получить то, что мне нужно, из обычного конвейерного потока из dplyr функций, но мне нужно абстрагировать это в функции NSE.Я предоставляю имя столбца для серии вызовов gather через аргумент ..., но это работает только с одним столбцом;Мне нужна опция использования нескольких столбцов.У меня проблемы с тем, как использовать quos(...) в этом случае.

Есть еще кое-что для функции, но я включаю только достаточно, чтобы показать ошибку.

Пример данных:

library(tidyverse)

race_pops <- structure(list(
    town = c("Hamden", "Hamden", "Hamden", "Hamden","New Haven", "New Haven", "New Haven", "New Haven", "West Haven","West Haven", "West Haven", "West Haven"), 
    race = c("Total","White", "Black", "Latino", "Total", "White", "Black", "Latino","Total", "White", "Black", "Latino"), 
    est = c(61476, 37043, 13209,6450, 130405, 40164, 42970, 37231, 54972, 28864, 10677, 10977), 
    moe = c(31, 1039, 998, 879, 60, 1395, 1383, 1688, 42, 1226,1119, 1032), 
    region = structure(c(1L, 1L, 1L, 1L, 2L, 2L, 2L,2L, 1L, 1L, 1L, 1L), .Label = c("Inner Ring", "New Haven"), class = "factor")), 
    class = c("tbl_df","tbl", "data.frame"), row.names = c(NA, -12L))

Вот рабочий бит, который выдает желаемый результат: 

race_pops %>%
    gather(key = measure, value = value, est, moe) %>%
    unite("grp2", race, measure, sep = "_") %>%
    spread(key = grp2, value = value) %>%
    gather(key = grp2, value = value, -town, -region, -starts_with("Total")) %>%
    head(10)
#> # A tibble: 10 x 6
#>    town       region     Total_est Total_moe grp2       value
#>    <chr>      <fct>          <dbl>     <dbl> <chr>      <dbl>
#>  1 Hamden     Inner Ring     61476        31 Black_est  13209
#>  2 New Haven  New Haven     130405        60 Black_est  42970
#>  3 West Haven Inner Ring     54972        42 Black_est  10677
#>  4 Hamden     Inner Ring     61476        31 Black_moe    998
#>  5 New Haven  New Haven     130405        60 Black_moe   1383
#>  6 West Haven Inner Ring     54972        42 Black_moe   1119
#>  7 Hamden     Inner Ring     61476        31 Latino_est  6450
#>  8 New Haven  New Haven     130405        60 Latino_est 37231
#>  9 West Haven Inner Ring     54972        42 Latino_est 10977
#> 10 Hamden     Inner Ring     61476        31 Latino_moe   879

Это функция, до которой я получаю ошибку: 

gather_grp <- function(df, grp = group, value = est, moe = moe, ...) {
    name_vars <- quos(...)
    grp_var <- enquo(grp)
    value_var <- enquo(value)
    moe_var <- enquo(moe)

    df %>%
        gather(key = measure, value = value, -(!!!name_vars), -(!!grp_var)) %>%
        unite("grp2", !!grp_var, measure, sep = "_") %>%
        spread(key = grp2, value = value) %>%
        gather(key = grp2, value = value, -(!!!name_vars), -starts_with("Total"))
}

Функция работает, если я опускаю region и использую только один столбец town: 

race_pops %>%
    select(-region) %>%
    gather_grp(grp = race, value = est, moe = moe, town) %>%
    head(10)
#> # A tibble: 10 x 5
#>    town       Total_est Total_moe grp2       value
#>    <chr>          <dbl>     <dbl> <chr>      <dbl>
#>  1 Hamden         61476        31 Black_est  13209
#>  2 New Haven     130405        60 Black_est  42970
#>  3 West Haven     54972        42 Black_est  10677
#>  4 Hamden         61476        31 Black_moe    998
#>  5 New Haven     130405        60 Black_moe   1383
#>  6 West Haven     54972        42 Black_moe   1119
#>  7 Hamden         61476        31 Latino_est  6450
#>  8 New Haven     130405        60 Latino_est 37231
#>  9 West Haven     54972        42 Latino_est 10977
#> 10 Hamden         61476        31 Latino_moe   879

Но я не могу предоставить town и region для ...: 

race_pops %>%
    gather_grp(grp = race, value = est, moe = moe, town, region)
#> Error in (~town): 2 arguments passed to '(' which requires 1

Создано в 2018-05-08 пакетом Представить (v0.2.0).

Заранее спасибо!

1 Ответ

0 голосов
/ 08 мая 2018

Мы можем обернуть c, и оно должно работать 

gather_grp <- function(df, grp = group, value = est, moe = moe, ...) {
    name_vars <- quos(...)
    grp_var <- enquo(grp)
    value_var <- enquo(value)
    moe_var <- enquo(moe)


    df %>%
        gather(key = measure, value = value, -c(!!!name_vars), -!!grp_var) %>%
        unite("grp2", !!grp_var, measure, sep = "_") %>%
        spread(key = grp2, value = value) %>%
        gather(key = grp2, value = value, -c(!!!name_vars), -starts_with("Total"))
}

- запуск функции 

race_pops %>%
    gather_grp(grp = race, value = est, moe = moe, town, region)
# A tibble: 18 x 6
#   town       region     Total_est Total_moe grp2       value
#   <chr>      <fct>          <dbl>     <dbl> <chr>      <dbl>
# 1 Hamden     Inner Ring     61476        31 Black_est  13209
# 2 New Haven  New Haven     130405        60 Black_est  42970
# 3 West Haven Inner Ring     54972        42 Black_est  10677
# 4 Hamden     Inner Ring     61476        31 Black_moe    998
# 5 New Haven  New Haven     130405        60 Black_moe   1383
# 6 West Haven Inner Ring     54972        42 Black_moe   1119
# 7 Hamden     Inner Ring     61476        31 Latino_est  6450
# 8 New Haven  New Haven     130405        60 Latino_est 37231
# 9 West Haven Inner Ring     54972        42 Latino_est 10977
#10 Hamden     Inner Ring     61476        31 Latino_moe   879
#11 New Haven  New Haven     130405        60 Latino_moe  1688
#12 West Haven Inner Ring     54972        42 Latino_moe  1032
#13 Hamden     Inner Ring     61476        31 White_est  37043
#14 New Haven  New Haven     130405        60 White_est  40164
#15 West Haven Inner Ring     54972        42 White_est  28864
#16 Hamden     Inner Ring     61476        31 White_moe   1039
#17 New Haven  New Haven     130405        60 White_moe   1395
#18 West Haven Inner Ring     54972        42 White_moe   1226

Для случая с одним столбцом нам нужно select выделить «регион» или «город», так как он также будет столбцом в наборе данных (или его необходимо изменить в функции) 

race_pops %>% 
    dplyr::select(-region) %>% 
    gather_grp(grp = race, value = est, moe = moe, town)
...