Как автоматически обращаться к именам расширенных столбцов - dplyr

Question

С примером прямо из справки:

stocks <- data.frame(
  time = as.Date('2009-01-01') + 0:9,
  X = rnorm(10, 0, 1),
  Y = rnorm(10, 0, 2),
  Z = rnorm(10, 0, 4)
)
stocksm <- stocks %>% gather(stock, price, -time)
stocksm %>% spread(stock, price)

Теперь я хотел бы создать скорректированные копии столбцов X, Y, Z, которые создаются с использованием spread. Я мог бы просто пойти на

stocksm %>% spread(stock, price) %>% mutate_at(vars(X, Y, Z), ~. + 1)

но это требует указания имен вручную, и, поскольку я хотел бы заблокировать все эти операции в функции, мне нужен какой-то способ передать имена вновь созданных столбцов. Я также не могу ссылаться по столбцу ID, так как количество столбцов должно быть произвольным Как мне это сделать?

Answer 1

Как насчет этого:

stocksm %>% {
  variables <- unique(.$stock)
  spread(., stock, price) %>% mutate_at(vars(variables), ~. + 1)
}

Но, честно говоря, я не могу представить ни одного примера, где было бы полезно распространяться первым.

Answer 2

Вы можете сделать это следующими способами:

Скажите mutate_at, чтобы взять все столбцы, кроме time:

stocksm %>% 
  spread(stock, price) %>%
  mutate_at(setdiff(names(.),"time"), ~. + 1)

"Взломать" mutate_all сначала группировкой, поскольку она игнорирует переменную группировки, а ваше действие не зависит от группы:

stocksm %>% 
  spread(stock, price) %>% 
  group_by(time) %>%
  mutate_all(~. + 1) %>% 
  ungroup

выход

# A tibble: 10 x 4
         time      X      Y      Z
       <date>  <dbl>  <dbl>  <dbl>
 1 2009-01-01  0.180  0.910  0.775
 2 2009-01-02  1.487  0.968  0.377
 3 2009-01-03  1.738  2.888 -4.883
 4 2009-01-04  1.576  2.642 -0.913
 5 2009-01-05  0.695  2.188  2.672
 6 2009-01-06  2.512  2.838  6.435
 7 2009-01-07  1.390  2.564  0.589
 8 2009-01-08  0.379  1.149  2.551
 9 2009-01-09 -1.215 -2.979  0.785
10 2009-01-10  2.125  2.240 -4.508

Answer 3

library(tidyr)
library(dplyr)

stocksm %>% 
  mutate_at(vars(stock), paste0, ".spread") %>% 
  spread(stock, price) %>% 
  mutate_at(vars(ends_with(".spread")), ~ . + 1) %>% 
  rename_at(vars(ends_with(".spread")), sub, 
            pattern = "\\.spread$", replacement = "")

#>          time           X            Y         Z
#> 1  2009-01-01  1.41134222  0.799908952 -2.547349
#> 2  2009-01-02 -0.03308714  1.510462889  2.909370
#> 3  2009-01-03  0.60722022 -1.725101030  4.697402
#> 4  2009-01-04  1.76681942 -2.538483592 -1.051576
#> 5  2009-01-05  0.05529327 -1.435663427  2.904005
#> 6  2009-01-06  2.69963414  2.364903583  3.957252
#> 7  2009-01-07  1.54475128 -4.289821346  3.043676
#> 8  2009-01-08  2.05165925  3.219704006 13.895450
#> 9  2009-01-09  0.05304517  0.008986952  5.257769
#> 10 2009-01-10  2.48281523  1.336480065  5.212230