嵌套数据帧的函数向量算法

时间:2018-05-18 17:00:12

标签: r dplyr purrr broom

我想创建一个返回向量的函数 - 向量需要包含另一个向量中的值的总和(monthly_amount),但只将“&_99; monthly_amount”中的值相加。相应的' year_month' (第三个向量)到函数返回其值的索引。

我保证在看到数据之后会有一点意义......

这是我的数据:

我有一个包含89个嵌套数据框的数据框(使用tidyr存储在第三列中),它看起来像这样:

 head(df)

    # A tibble: 6 x 5
  industry location data                
     <dbl>    <dbl> <list>              
1       8.       9. <tibble [627 × 5]>  
2       1.       4. <tibble [5,879 × 5]>
3       9.       2. <tibble [1,271 × 5]>
4       1.       2. <tibble [8,939 × 5]>
5       1.       5. <tibble [6,093 × 5]>
6      10.       5. <tibble [315 × 5]> 

第三列(&#39;数据&#39;)的内容结构相同,看起来像这样:

> df$data[[1]]

    # A tibble: 627 x 5
   date       monthly_amount year  month year_month
   <date>              <dbl> <chr> <chr> <chr>     
 1 2013-01-01        753851. 2013  01    2013_01   
 2 2013-01-01        154426. 2013  01    2013_01   
 3 2013-01-01        499604. 2013  01    2013_01   
 4 2013-01-01        379321. 2013  01    2013_01   
 5 2013-01-01        213702. 2013  01    2013_01   
 6 2013-01-01        274118. 2013  01    2013_01   
 7 2013-01-01        282391. 2013  01    2013_01   
 8 2013-01-01        236070. 2013  01    2013_01   
 9 2013-01-01        182512. 2013  01    2013_01   
10 2013-01-01        428778. 2013  01    2013_01   

这是我想要实现的一个虚拟例子:

# A tibble: 627 x 5
   date       monthly_amount year  month year_month  desired_outcome
   <date>              <dbl> <chr> <chr> <chr>                <dbl> 
 1 2013-01-01        753851. 2013  01    2013_01 "sum of all monthly_amount     
 2 2013-01-01        154426. 2013  01    2013_01 where year_month==2013_01"
 3 2013-01-02        499604. 2013  02    2013_02 "sum of all monthly_amount  
 4 2013-01-02        379321. 2013  02    2013_02 where year_month==2013_02"  
 5 2013-01-02        213702. 2013  02    2013_02   
 6 2013-01-03        274118. 2013  03    2013_03   etc...
 7 2013-01-03        282391. 2013  03    2013_03   
 8 2013-01-04        236070. 2013  04    2013_04   
 9 2013-01-04        182512. 2013  04    2013_04   
10 2013-01-04        428778. 2013  04    2013_04 

理想情况下,我会将此函数映射到所有89个数据帧,因此它需要非常通用,而不是更精确的for循环。

非常感谢有关如何更好地表达这个问题(或解决此问题!)的任何建议。

1 个答案:

答案 0 :(得分:1)

我们可以使用data遍历map,然后按'year_month'分组,将'desired_outcome'创建为'monthly_amount'的sum

df %>%
    mutate(data = map(data, ~ .x %>%
                               group_by(year_month) %>%
                               mutate(desired_outcome = sum( monthly_amount))))