使用dplyr从名称中提取列表

Pav*_*ive 3 r forecasting dplyr data.table

我有一个数据表,其中包含许多产品的日期消耗.我生成了每个产品的预测,现在想要获得平均值,并在期间+1上限80%.问题是预测对象是一个具有不同结构的列表,具体取决于所使用的方法,因此我无法通过索引来检索值(我可以通过名称来检索data.table).

这是(虚拟)数据和代码:

# load required libraries
library(data.table)
library(xts)
library(forecast)
library(dplyr)

# create random data
set.seed(1)
a <- data.table(prod = sample(LETTERS[1:5], 20, TRUE), cons = sample(1:50, 20, TRUE), dt = sample(seq(as.Date("2016/06/01"), as.Date("2016/07/27"), by = "day"), 20, FALSE))

# create a time series of purchases
b <- a[, .(C=sum(cons)), by = .(dt, prod)][, x := .(list(xts(x = C, order.by = dt))), by = prod]
b <- b[, .SD[1,], by = prod]

# create a "reference" timeseries
dts <- xts(order.by = seq(as.Date("2016/06/01"), as.Date("2016/07/27"), by = "day"))

# merge reference and calculated timeseries, so zeros appear
b[, x2 := .(list(merge.xts(dts, x[[1]], fill = 0))), by = prod]

# calculate forecast for each extended timeseries
b[, fc := .(list(forecast(x2[[1]]))), by = prod]
Run Code Online (Sandbox Code Playgroud)

现在我想提取平均值和上限.问题是平均值有时位于列表的第2个插槽中,有时不在,所以我必须按名称调用它.在data.table我这样做:

b[, mn := fc[[1]]$mean[1], by = prod]
b[, up := fc[[1]]$upper[1,1], by = prod]
Run Code Online (Sandbox Code Playgroud)

但如果我尝试做同样的事情dplyr,那么我会收到一个关闭错误:

b  %>% mutate(mnD = .$fc[[1]]$mean[1])
## Error: invalid subscript type 'closure'

b  %>% mutate(mnD = fc[[1]]$mean[1])
## Error: invalid subscript type 'closure'
Run Code Online (Sandbox Code Playgroud)

我做错了什么,我怎么能做到这一点dplyr

Sim*_*son 5

与结合map_dbl()purrr包将工作如下:

library(dplyr)
library(purrr)
b %>% as_data_frame() %>% mutate(mnD = map_dbl(fc, ~ .$mean[1]))
#> # A tibble: 5 x 7
#>    prod         dt     C         x        x2             fc       mnD
#>   <chr>     <date> <int>    <list>    <list>         <list>     <dbl>
#> 1     B 2016-07-17    47 <S3: xts> <S3: xts> <S3: forecast> 2.5241999
#> 2     C 2016-07-14    33 <S3: xts> <S3: xts> <S3: forecast> 1.1749266
#> 3     E 2016-06-30     7 <S3: xts> <S3: xts> <S3: forecast> 0.5952119
#> 4     D 2016-06-24    20 <S3: xts> <S3: xts> <S3: forecast> 3.3695962
#> 5     A 2016-07-04    18 <S3: xts> <S3: xts> <S3: forecast> 0.8421001
Run Code Online (Sandbox Code Playgroud)

除此之外,as_data_frame()不需要,但添加以完整的方式打印结果.没有它,列表列将打印所有数据.

  • 没有必要,只是这样打印得当.没有它,列表列将完整打印(而不是<S3:xxx>).但好的提醒 - 我会编辑提到这一点. (2认同)