Question

我有以下学生记录数据框。我要确定的是在9年级时第一次参加2014年某个课程的学生。

names.first<-c('a','a','b','b','c','d')
names.last<-c('c','c','z','z','f','h')
year<-c(2014,2013,2014,2015,2015,2014)
grade<-c(9,8,9,10,10,10)

df<-data.frame(names.first,names.last,year,grade)
df

为此，我使用以下语句来表示我希望学生在year==2014程序及其grade ==9.程序中学习

 df$first.cohort<-ifelse(df$year==2014 & df$grade==9,1,0)
df



  names.first names.last year grade first.cohort
1           a          c 2014     9            1
2           a          c 2013     8            0
3           b          z 2014     9            1
4           b          z 2015    10            0
5           c          f 2015    10            0
6           d          h 2014    10            0

但是，您可能会注意到，其中包括2014年未参加该计划的学生，例如2013年开始的a学生。我如何创建ifelse语句，只捕获那些在9年级就读，并于2014年首次启动该计划，因此df看起来像

  names.first names.last year grade first.cohort
1           a          c 2014     9            0
2           a          c 2013     8            0
3           b          z 2014     9            1
4           b          z 2015    10            0
5           c          f 2015    10            0
6           d          h 2014    10            0

Answer 1

在first按“名称”和“年份”命名后，我们可以使用arrange创建逻辑表达式

library(dplyr)
df %>% 
   arrange(names, year) %>% 
   group_by(names) %>% 
   mutate(first.cohort = as.integer(grade == 9 & first(year) == 2014))
# A tibble: 6 x 4
# Groups:   names [4]
#  names  year grade first.cohort
#  <fct> <dbl> <dbl>        <int>
#1 a      2013     8            0
#2 a      2014     9            0
#3 b      2014     9            1
#4 b      2015    10            0
#5 c      2015    10            0
#6 d      2014    10            0

为了保持与输入数据集相同的顺序，我们可以先创建一个序列列，然后在arrange之后的列上进行mutate

df %>% 
   mutate(rn = row_number()) %>%
   arrange(names, year) %>% 
   group_by(names) %>% 
   mutate(first.cohort = as.integer(grade == 9 & first(year) == 2014)) %>%
   ungroup %>%
   arrange(rn) %>%
   select(-rn)

或者与data.table一起使用相同的逻辑，这些逻辑还有一个额外的优点，就是保持与输入数据集中的顺序相同

library(data.table)
setDT(df)[order(names, year), first.cohort := as.integer(grade == 9 &
           first(year) == 2014), names]

更新

使用OP帖子中的新示例，我们通过“名称”列进行分组

df %>% 
   arrange(names.first, names.last, year) %>%
   group_by(names.first, names.last) %>%
   mutate(first.cohort = as.integer(grade == 9 & first(year) == 2014))
# A tibble: 6 x 5
# Groups:   names.first, names.last [4]
#  names.first names.last  year grade first.cohort
#  <fct>       <fct>      <dbl> <dbl>        <int>
#1 a           c           2013     8            0
#2 a           c           2014     9            0
#3 b           z           2014     9            1
#4 b           z           2015    10            0
#5 c           f           2015    10            0
#6 d           h           2014    10            0

Answer 2

使用dplyr

library(dplyr)
df%>%group_by(names)%>%dplyr::mutate(Fc=as.numeric((year==2014&grade==9)&(min(year)==2014)))
# A tibble: 6 x 4
# Groups:   names [4]
   names  year grade    Fc
  <fctr> <dbl> <dbl> <dbl>
1      a  2014     9     0
2      a  2013     8     0
3      b  2014     9     1
4      b  2015    10     0
5      c  2015    10     0
6      d  2014    10     0

如何确定元素首次出现的行？

2 个答案:

更新