获取data.table中前一个组的最后一行

Question

这就是我的数据表的样子:

library(data.table)
dt <- fread('
    Product  Group    LastProductOfPriorGroup
    A          1          NA
    B          1          NA
    C          2          B
    D          2          B
    E          2          B
    F          3          E
    G          3          E
')

该LastProductOfPriorGroup列是我想要的列.我正在尝试从前一组的最后一行获取产品.所以在前两行中,没有先前的组,因此它是NA.在第三行中,前一组1的最后一行中的产品是B.我试图通过这个来实现这个目标

dt[,LastGroupProduct:= shift(Product,1), by=shift(Group,1)]

无济于事.

Answer 1

你可以做到

dt[, newcol := shift(dt[, last(Product), by = Group]$V1)[.GRP], by = Group]

这会导致以下更新dt,其中newcol您的所需列与不必要的长名称匹配.;)

   Product Group LastProductOfPriorGroup newcol
1:       A     1                      NA     NA
2:       B     1                      NA     NA
3:       C     2                       B      B
4:       D     2                       B      B
5:       E     2                       B      B
6:       F     3                       E      E
7:       G     3                       E      E

让我们从内到外打破代码.我将...用来表示累积的代码:

• dt[, last(Product), by = Group]$V1 将每组中的最后一个值作为字符向量.
• shift(...) 在前一次调用中移动字符向量
• dt[, newcol := ...[.GRP], by = Group]分组依据Group并使用内部.GRP值进行索引

更新:弗兰克为我的代码提出了一个很好的观点,一遍又一遍地计算每个组的班次.为避免这种情况,我们可以使用其中之一

shifted <- shift(dt[, last(Product), Group]$V1)
dt[, newcol := shifted[.GRP], by = Group]

这样我们就不计算每个组的班次.或者,我们可以在评论中采纳弗兰克的好建议并执行以下操作.

dt[dt[, last(Product), by = Group][, v := shift(V1)], on="Group", newcol := i.v] 

Answer 2

另一种方法是将最后一个组的值保存在变量中.

this = NA_character_    # initialize
dt[, LastProductOfPriorGroup:={ last<-this; this<-last(Product); last }, by=Group]
dt
   Product Group LastProductOfPriorGroup
1:       A     1                      NA
2:       B     1                      NA
3:       C     2                       B
4:       D     2                       B
5:       E     2                       B
6:       F     3                       E
7:       G     3                       E

NB:last()是一个data.table函数,它返回向量的最后一项(在本例中为Product列).

这也应该很快,因为没有调用逻辑来获取最后一个组的值; 它只依赖于按顺序运行的组(他们这样做).