删除r中每个因素级别的最后2行

发布于01月17日

我有一个数据帧，其中一个因子具有不同数量的行数.我想删除每个因素级别的最后2行. test<-data.frame(id=c(1,1,1,1,1,2,2,2,2,2,2,3,3,3,3,3,3,3), val=c(3,5,4,6,7,4,1,6,7,8,4,2,0,3,6,8,1,2), trt=c(1,1,1,1,1,1,1,1,1,1,1,2,2,2,2,2,2,2))个

结果应该是这样的:

   id val trt
1   1   3   1
2   1   5   1
3   1   4   1
4   2   4   1
5   2   1   1
6   2   6   1
7   2   7   1
8   3   2   2
9   3   0   2
10  3   3   2
11  3   6   2
12  3   8   2

另外，我还需要在数据框中保留各种其他列.我编辑了示例df以包括另一个与本练习无关的因子，但我确实需要它(以及更多)用于将来的计算.

推荐答案

使用data.table:

library(data.table)

setDT(test)[,.(val = val[1:(.N - 2)]), id][]
#>     id val
#>  1:  1   3
#>  2:  1   5
#>  3:  1   4
#>  4:  2   4
#>  5:  2   1
#>  6:  2   6
#>  7:  2   7
#>  8:  3   2
#>  9:  3   0
#> 10:  3   3
#> 11:  3   6
#> 12:  3   8

或者，如果有其他列，则使用.SD:

setDT(test)[,.SD[1:(.N - 2)], id][]
#>     id val trt
#>  1:  1   3   1
#>  2:  1   5   1
#>  3:  1   4   1
#>  4:  2   4   1
#>  5:  2   1   1
#>  6:  2   6   1
#>  7:  2   7   1
#>  8:  3   2   2
#>  9:  3   0   2
#> 10:  3   3   2
#> 11:  3   6   2
#> 12:  3   8   2

请注意，如果数据尚未按id分组，这将相对于原始数据重新排列数据.

基数R(如果test按id排序):

test[sequence((n <- rle(test$id)[[1]]) - 2, c(1, cumsum(n[-length(n)]) + 1)),]
#>     id val trt
#>  1:  1   3   1
#>  2:  1   5   1
#>  3:  1   4   1
#>  4:  2   4   1
#>  5:  2   1   1
#>  6:  2   6   1
#>  7:  2   7   1
#>  8:  3   2   2
#>  9:  3   0   2
#> 10:  3   3   2
#> 11:  3   6   2
#> 12:  3   8   2