如果只有第一个观察符合一个标准，有没有办法子集一个组？

Question

ID = c(1,1,1,2,2,2,3,3,3,4,4,4,4)
VAR_1 = c(2,4,6,1,7,9,4,4,3,1,7,4,0)
VAR_2 = c(NA,NA,NA,NA,NA,20190101,20190101,20190101,NA,20190101,NA,NA,NA)
df2 = data.frame(ID,VAR_1,VAR_2)

df                                       df_subset   

ID VAR_1 VAR_2                           ID VAR_1 VAR_2
1   2     NA                             3   4     20190101                           
1   4     NA                             3   4     20190101
1   6     NA                             3   3     NA
2   1     NA                             4   1     20190101
2   7     NA                             4   7     NA
2   9     20190101                       4   4     NA
3   4     20190101                       4   0     NA
3   4     20190101
3   3     NA
4   1     20190101
4   7     NA
4   4     NA
4   0     NA

我通过几个步骤完成了这一过程（我对原始数据进行了子集，只按组进行第一次观察，给VAR_1赋一个特殊值，重新合并，然后最后按特殊值进行过滤），但我想知道是否有更简单、更优雅（可能也是）更有效的方法。我不需要VAR_1，所以如果需要，可以更改VAR_1，以提供更快的解决方案。

如有任何帮助，不胜感激。

Answer 1

使用dplyr，我们可以group_byID，并且仅当每个组中的第一个值是非na时才选择组。

library(dplyr)

df2 %>%
  group_by(ID) %>%
  filter(!is.na(VAR_2[1L]))

#     ID VAR_1    VAR_2
#  <dbl> <dbl>    <dbl>
#1     3     4 20190101
#2     3     4 20190101
#3     3     3       NA
#4     4     1 20190101
#5     4     7       NA
#6     4     4       NA
#7     4     0       NA

提取第一个值的一些变体可以是（感谢@tmfmnk)

df2 %>%  group_by(ID) %>% filter(!is.na(first(VAR_2)))

或

df2 %>% group_by(ID) %>%  filter(!is.na(nth(VAR_2, 1)))

df2[with(df2, ave(!is.na(VAR_2), ID, FUN = function(x) x[1L])), ]

subset(df2, ID %in% names(na.omit(sapply(split(df2$VAR_2, df2$ID), head, 1))))