Как передать список переменных для выбора в tq_mutate (TidyQuant) из функции? - PullRequest
Новая
       84

Как передать список переменных для выбора в tq_mutate (TidyQuant) из функции?

0 голосов
/ 19 декабря 2018

Как я могу изменить свой код, чтобы я мог передать список переменных в select в tq_mutate при вызове из функции?

Следующий код работает при вызове вне функции, ноне при вызове внутри функции:

(отредактировано для добавления воспроизводимого примера): 

library(tidyquant)

# WORKS
x <- data_frame(date=seq(as.Date('2018-01-01'),as.Date('2018-01-10'),by=1), a=seq(1,10,by=1), b=seq(11,20,by=1))
k=c(1,2)

lag_cols <- c('a', 'b')

temp_names <- crossing(lag_cols, k)
temp_names <- temp_names %>% arrange(k, lag_cols)
col_names <- paste0(temp_names$lag_cols, '_lag', temp_names$k)

result <- x %>% tq_mutate(
    select = c('a', 'b'),
    mutate_fun = lag.xts,
    k = k,
    col_rename=col_names)

result

# DOESN'T WORK
lag_data <- function(df, k) {
  lag_cols_in <- c('a', 'b')

  temp_names <- crossing(lag_cols_in, k)
  temp_names <- temp_names %>% arrange(k, lag_cols_in)
  col_names <- paste0(temp_names$lag_cols_in, '_lag', temp_names$k)

  df %>% tq_mutate(
      select = lag_cols_in,
      mutate_fun = lag.xts,
      k = k,
      col_rename=col_names)
}

Мое чтение показывает, что tq_mutate использует select_ при вызове внутри функции, и поэтому мне нужнопередать .dots=c('a','b','c') вызов select.

Как я могу изменить свой код, чтобы я мог передать список переменных?Обратите внимание, что помещение select=c('a', 'b') внутри вызова tq_mutate работает в обоих случаях, поэтому я предполагаю, что это как-то связано со средами и областью видимости переменной.

Вот сообщение об ошибке: 

Error in .f(.x[[i]], ...) : object 'lag_cols_in' not found
23.
.f(.x[[i]], ...)
22.
map(.x[sel], .f, ...)
21.
map_if(quos, !is_helper, eval_tidy, mask)
20.
vars_select_eval(.vars, quos)
19.
tidyselect::vars_select(names(.data), !!!quos(...))
18.
select.data.frame(.data, !!!dots)
17.
select(.data, !!!dots)
16.
select_.data.frame(data, select)
15.
dplyr::select_(data, select)
14.
tq_transmute_.tbl_df(data = data, select = select, mutate_fun = mutate_fun, col_rename = col_rename, ... = ...)
13.
tq_transmute_(data = data, select = select, mutate_fun = mutate_fun, col_rename = col_rename, ... = ...)
12.
tq_mutate_.tbl_df(data = data, select = lazyeval::expr_text(select), mutate_fun = lazyeval::expr_text(mutate_fun), col_rename = col_rename, ... = ...)
11.
tq_mutate_(data = data, select = lazyeval::expr_text(select), mutate_fun = lazyeval::expr_text(mutate_fun), col_rename = col_rename, ... = ...)
10.
tq_mutate(., select = lag_cols_in, mutate_fun = lag.xts, k = k, col_rename = col_names)
9.
function_list[[k]](value)
8.
withVisible(function_list[[k]](value))
7.
freduce(value, `_function_list`)
6.
`_fseq`(`_lhs`)
5.
eval(quote(`_fseq`(`_lhs`)), env, env)
4.
eval(quote(`_fseq`(`_lhs`)), env, env)
3.
withVisible(eval(quote(`_fseq`(`_lhs`)), env, env))
2.
df %>% tq_mutate(select = lag_cols_in, mutate_fun = lag.xts, k = k, col_rename = col_names)
1.
lag_data(x, k)

1 Ответ

0 голосов
/ 20 декабря 2018

Возможно, есть лучший способ, но, похоже, он работает ... 

library(tidyquant)
x <- data_frame(date=seq(as.Date('2018-01-01'),as.Date('2018-01-10'),by=1), a=seq(1,10,by=1), b=seq(11,20,by=1))
k <- c(1,2)
lag_data <- function(df, k) {
  lag_cols_in <- c('a', 'b')
  temp_names <- crossing(lag_cols_in, k)
  temp_names <- temp_names %>% arrange(k, lag_cols_in)
  col_names <- paste0(temp_names$lag_cols_in, '_lag', temp_names$k)

  df %>% tq_mutate(
      select = lag_cols_in[1:length(lag_cols_in)],
      mutate_fun = lag.xts,
      k = k,
      col_rename = col_names)
}

lag_data(x,k)
...