每个 ID 保留一定数量的行

Question

我想为每个变量保留一定数量的行： 比如假设我有如下数据集（实际数据集比较大）：

part<-c(1,1,1,2,2,2,3,3)
choice<-c(6,7,15,42,1,1,4,5)
data<-cbind(part,choice)
des_data<-data[c(1,2,4,5,7,8),]

      part choice
[1,]    1      6
[2,]    1      7
[3,]    1     15
[4,]    2     42
[5,]    2      1
[6,]    2      1
[7,]    3      4
[8,]    3      5

我想只保留每个“部分”（参与者）的前 2 个观察结果。

这样最终的数据是这样的：

      part choice
[1,]    1      6
[2,]    1      7
[3,]    2     42
[4,]    2      1
[5,]    3      4
[6,]    3      5

我该怎么做？

谢谢！

Answer 1

library(dplyr)
data %>%
  as.data.frame %>% #(your example code creates a matrix)
  group_by(part) %>%
  slice(1:2)

Answer 2

data.table 的选项

library(data.table)
as.data.table(data)[, .SD[1:2], part]
#    part choice
#1:    1      6
#2:    1      7
#3:    2     42
#4:    2      1
#5:    3      4
#6:    3      5