在组后插入一行并填充下一行值答案

【问题标题】：insert a row after a group and fill with next row values在组后插入一行并填充下一行值
【发布时间】：2018-12-31 12:08:53
【问题描述】：

我想在每个组 (grp) 之后插入一个新行，并用下一行的值和前一行的值填充一些新列

我正在尝试使用：

x<-rbind(setDT(DF), DF[,.SD[.N], grp][, color := shift(color,1L, type = "lag")][, Lat:= shift(Lat,1L, type = "lead")])[order(id)]

在 DF 上：

a <- c(1,2,3,4,5,6,7,8,9,10)
b <- c(10,20,30,40,50,60,70,80,90,100)
c <- c("a","a","b","b","b","a","a","b","c","c")
d <- c(11,23,67,89,90,100,101,123,200,290)
df <- data.frame(color=a, Lat=b, grp=c, id=d)

我可能使用了不正确的 shift()，我似乎无法让它工作。

谢谢

预期结果是：

   color Lat grp  id
1      1  10   a  11
2      2  20   a  23
new row with color from previous row, Lat from next row, grp from previous and if from next
3      3  30   b  67
4      4  40   b  89
5      5  50   b  90
new row as before
6      6  60   a 100
7      7  70   a 101
new row as before
8      8  80   b 123
new row as before
9      9  90   c 200
10    10 100   c 290

【问题讨论】：

你能显示预期的输出吗？当前代码有什么问题
修改了预期的结果。
完美！正是我想要的。不认为我理解代码虽然

标签： r list insert row shift

【解决方案1】：

在我们按“grp”的rleid 进行分组之前，我们创建了两个lead 列，即“Lat”和“id”的下一行值。在这里，rleid 正在检查 'grp' 的相邻元素是否相同。如果不同，则为该元素分配一个新的 id

library(data.table)
setDT(df)[, c("LatN", "idN") := shift(.SD, type = 'lead'), .SDcols = c('Lat', 'id')]

通过'grp'的rleid获取所选列的last观察

tmp <- df[, .(color = color[.N], Lat = LatN[.N], id = idN[.N], grp = grp[.N]), 
       .(grp1 = rleid(grp))][, grp1 := NULL]

rbind 与原始数据集列，order 通过“颜色”并删除任何 NA 行

na.omit(rbind(df[,.(color, Lat, id, grp)], tmp)[order(color)])
#     color Lat  id grp
# 1:     1  10  11   a
# 2:     2  20  23   a
# 3:     2  30  67   a
# 4:     3  30  67   b
# 5:     4  40  89   b
# 6:     5  50  90   b
# 7:     5  60 100   b
# 8:     6  60 100   a
# 9:     7  70 101   a
#10:     7  80 123   a
#11:     8  80 123   b
#12:     8  90 200   b
#13:     9  90 200   c
#14:    10 100 290   c

【讨论】：