r - 使用while循环根据R中的重复值创建一个新变量

Question

所以我有一组看起来像这样的数据:

group  ID
aa     123
ab     123
bb     345
bb     345
bb     999
bb     999
cc     567
cd     567

当 ID 相同时，组应等于该 ID 的第二个条目。所以修正后的数据应该是这样的:

group  ID   group2
aa     123  ab
ab     123  ab
bb     345  bb
bb     345  bb
bb     999  bb
bb     999  bb
cc     567  cd
cd     567  cd

我还需要创建一个新变量来存储正确的组。这是我一直在尝试的:

n <- 1 + (1:(as.numeric(nrow(data))))
l <- 1:(as.numeric(nrow(data)))
while (data[n,1] == data[l,1]) { data$group2 <- data[n,1] }

Answer 1

这是一个使用 data.table 的选项。按'ID'分组后，选择'group'的第二个观察值并将其分配(:=)给'group2'

library(data.table)
setDT(df1)[, group2 := group[2], ID]
df1
#   group  ID group2
#1:    aa 123     ab
#2:    ab 123     ab
#3:    bb 345     bb
#4:    bb 345     bb
#5:    bb 999     bb
#6:    bb 999     bb
#7:    cc 567     cd
#8:    cd 567     cd