Question

我目前有一个数据框，它取自按时间顺序发生的事件的数据馈送。我想在我的数据的每一行上添加一个新列，如果事件类型是1＆amp;，则对应于下一个事件的player_id。结果是1

例如

player_id <- c(12, 17, 26, 3)
event_type <- c(1, 3, 1, 10)
outcome <- c(1, 0, 1, 1)
df <- data.frame(player_id, event_type, outcome)
df
     player_id  event_type outcome
1        12          1       1
2        17          3       0
3        26          1       1
4         3         10       1

所以最终结果

      player_id event_type outcome next_player_id
1        12          1       1             17
2        17          3       0             NA
3        26          1       1              3
4         3         10       1             NA

非常感谢，因为我还处于r

的初学者阶段

Answer 1

我们可以将ifelse与lead

一起使用

library(dplyr)
res <- df %>% 
         mutate(next_player_id =  ifelse(outcome==1 & event_type == 1, 
                                                lead(player_id), NA))
res
#  player_id event_type outcome next_player_id
#1        12          1       1             17
#2        17          3       0             NA
#3        26          1       1              3
#4         3         10       1             NA

或base R transform和ifelse

transform(df, next_player_id = ifelse(outcome == 1 & event_type==1, 
                c(player_id[-1], NA), NA))

编辑：根据@ RHertel的评论添加了&条件

或者代替ifelse，我们可以在两步过程中执行此操作，即1）创建逻辑索引，2）基于我们分配新列的索引

i1 <- with(df, event_type == 1 & outcome ==1)
df$next_player_id[i1] <- c(df$player_id[-1], NA)[i1]

if行上的语句r

1 个答案: