如何在 R 中计算具有不同消耗水平的滚动总和？

Question

我正在尝试使用 R 来衡量一个人在重新配药时手头上已经有多少天的处方药供应，同时考虑到所有以前的处方药。例如，如果我有这个 table...

  member rx_id  fill_date    to_date days_supply
1      A     1 2018-10-01 2018-10-02           2
2      B     1 2016-11-07 2016-11-10           4
3      B     2 2016-11-07 2016-12-04          28
4      B     3 2016-11-08 2016-11-09           2
5      B     4 2016-11-10 2016-12-03          24

我希望得到以下输出

  member rx_id  fill_date    to_date days_supply_on_hand
1      A     1 2018-10-01 2018-10-02                   0
2      B     1 2016-11-07 2016-11-10                   0
3      B     2 2016-11-07 2016-12-04                   4
4      B     3 2016-11-08 2016-11-09                  30
5      B     4 2016-11-10 2016-12-03                  26

对于会员 B，当第二个脚本与第一个脚本在同一天填写时，该个人手头已经有 4 天的 RX。当第三个脚本完成时，个人距离第一个脚本还有 3 天，距离第二个脚本还有 27 天（总共 30 天）。当第四个脚本被填充时，第三个脚本被耗尽，但是距离第一个脚本还有 1 天，距离第三个脚本还有 25 天（总共 26 个）。

我知道如何在 dplyr 和 data.table 中进行滚动总计，但我不知道如何根据以前的个人记录考虑不同的消耗水平。下面是重新制作原始代码 table，在此先感谢您的任何建议！

structure(list(member = structure(c(1L, 2L, 2L, 2L, 2L), .Label = 
c("A", 
"B"), class = "factor"), rx_id = c(1, 1, 2, 3, 4), fill_date = 
structure(c(17805, 
17112, 17112, 17113, 17115), class = "Date"), to_date = 
structure(c(17806, 
17115, 17139, 17114, 17138), class = "Date"), days_supply = c(2, 
4, 28, 2, 24)), .Names = c("member", "rx_id", "fill_date", 
"to_date", 
"days_supply"), row.names = c(NA, -5L), class = "data.frame")

Answer 1

使用简单的循环

dt$days_supply_on_hand <- 0
for (a in unique(dt$member)) {
  I <- which(.subset2(dt,1) == a)
  flDate <- as.integer(.subset2(dt,3)[I])
  toDate <- as.integer(.subset2(dt,4)[I])
  V <- vapply(seq_along(I), function (k) sum(toDate[1:(k-1)] - flDate[k] + 1), numeric(1))
  dt$days_supply_on_hand[I] <- c(0,V[-1])
}
dt
  member rx_id  fill_date    to_date days_supply days_supply_on_hand
1      A     1 2018-10-01 2018-10-02           2                   0
2      B     1 2016-11-07 2016-11-10           4                   0
3      B     2 2016-11-07 2016-12-04          28                   4
4      B     3 2016-11-08 2016-11-09           2                  30
5      B     4 2016-11-10 2016-12-03          24                  26

其中dt是上面提供的数据框。（请注意，使用 .subset2 或 as.integer 是为了提高效率 - 可以更改它们以提高可读性）。

Answer 2

library(data.table)
dt = as.data.table(your_df) # or setDT to convert in place

# merge on relevant days, then compute sum of supply - days elapsed
dt[dt, on = .(member, fill_date <= fill_date, to_date >= fill_date, rx_id < rx_id), by = .EACHI,
   sum(days_supply, na.rm = T) - sum(i.fill_date - x.fill_date, na.rm = T)]
#   member  fill_date    to_date rx_id      V1
#1:      A 2018-10-01 2018-10-01     1  0 days
#2:      B 2016-11-07 2016-11-07     1  0 days
#3:      B 2016-11-07 2016-11-07     2  4 days
#4:      B 2016-11-08 2016-11-08     3 30 days
#5:      B 2016-11-10 2016-11-10     4 26 days

如何在 R 中计算具有不同消耗水平的滚动总和？

How do I calculate a rolling sum with variable levels of depletion in R?

r

overlap

dplyr

data.table

rolling-sum