尝试对多个数据帧重复相同的代码

时间:2019-03-27 01:30:06

标签: r loops lapply mutate repeat

我有下面的代码,该代码接受一个称为dft1的数据帧,然后生成一个称为dfb1的结果数据帧。我想对多个输入数据帧(例如dft1,dft2)重复相同的代码,所有输入数据帧都以一个数字向结尾索引,然后使用相同的模式(即dfb1,dfb2,...)存储结果。 我尝试了许多方法,例如使用dapply或for循环,但是鉴于内部代码的性质,我无法获得预期的结果。


#define the function for rolling

window <- 24
rolling_lm <- 
  rollify(.f = function(R_excess, MKT_RF, SMB, HML) {
    lm(R_excess ~ MKT_RF + SMB + HML)
  }, window = window, unlist = FALSE)

#rolling over the variable 

dfb1 <-
  dft1 %>% 
  mutate(rolling_ff = 
           rolling_lm(R_excess, 
                      MKT_RF, 
                      SMB, 
                      HML)) %>% 
  mutate(tidied = map(rolling_ff, 
                      tidy, 
                      conf.int = T)) %>% 
  unnest(tidied) %>% 
  slice(-1:-23) %>% 
  select(date, term, estimate, conf.low, conf.high) %>% 
  filter(term != "(Intercept)") %>% 
  rename(beta = estimate, factor = term) %>% 
  group_by(factor)

1 个答案:

答案 0 :(得分:1)

在函数中添加要应用于每个数据框的命令

apply_fun <- function(df) {
   df %>% 
    mutate(rolling_ff = 
         rolling_lm(R_excess, 
                    MKT_RF, 
                    SMB, 
                    HML)) %>% 
    mutate(tidied = map(rolling_ff, 
                    tidy, 
                    conf.int = T)) %>% 
    unnest(tidied) %>% 
    slice(-1:-23) %>% 
    select(date, term, estimate, conf.low, conf.high) %>% 
    filter(term != "(Intercept)") %>% 
    rename(beta = estimate, factor = term) %>% 
    group_by(factor)
}

现在将功能应用于每个数据框并将结果存储在列表中

n <- 10
out <- setNames(lapply(mget(paste0("dft", 1:n)), apply_fun), paste0("dfb", 1:n))

假设您有dft1dft2之类的输入数据帧...这将输出一个数据帧列表,您现在可以通过out[['dfb1']]out[['dfb2']]等访问这些数据帧。根据您拥有的n个数据帧的数量来更改dft的值。

如果列表中已经存在数据,我们可以通过这样做来避免mget

setNames(lapply(result, apply_fun), paste0("dfb", 1:n))