如果列的第二行与 r 中的值匹配，则创建新变量

Question

我想创建一个新变量 A_row_error，如果 first_A_row = TRUE 和下一行的消息 = ERROR（如果下一行 A_row_error = TRUE 是任何其他消息，则为 FALSE）。请注意，下一行由时间戳确定，并且必须按升序排列。 A_row_error 的 TRUE 消息也必须与 first_A_row = TRUE 位于同一行（有关所需输出，请参见下表）。

样本数据集：

participant_id <- c("ps1", "ps1", "ps1", "ps1", "ps2", "ps2", "ps3", "ps3", "ps3", "ps3")
timestamp <- c(0.01, 0.02, 0.03, 0.04, 0.01, 0.02, 0.01, 0.02, 0.03, 0.04)
event <- c("A", "A", "A", "B", "B", "A", "A", "A", "B", "A")
first_A_row <- c("TRUE", "FALSE", "FALSE", "FALSE", "FALSE", "TRUE", "TRUE", "FALSE", "FALSE", "FALSE")
message <- c("ACCEPTED", "ERROR", "DECLINED", "ACCEPTED", "HELLO", "BYE", "ACCEPTED", "BYE", "ERROR", "ACCEPTED")
data.frame(participant_id, timestamp, event, first_A_row, message)

期望的输出：

participant_id	timestamp	event	first_A_row	message	A_row_error
ps1	0.01	A	TRUE	ACCEPTED	TRUE
ps1	0.02	A	FALSE	ERROR	FALSE
ps1	0.03	A	FALSE	DECLINED	FALSE
ps1	0.04	B	FALSE	ACCEPTED	FALSE
ps2	0.01	B	FALSE	HELLO	FALSE
ps2	0.02	A	TRUE	BYE	FALSE
ps3	0.01	A	TRUE	ACCEPTED	FALSE
ps3	0.02	A	FALSE	BYE	FALSE
ps3	0.03	B	FALSE	ERROR	FALSE
ps3	0.04	A	FALSE	ACCEPTED	FALSE

Answer 1

我们可以通过 'participant_id' 进行分组，并使用 'message' 的 lead 创建逻辑，即如果下一个值为 'ERROR' 并且 'first_A_row' 的给定行为 TRUE（@987654322 列@ 被创建为字符而不是逻辑 - 因此它被转换为 as.logical)

library(dplyr)
df1 %>% 
   group_by(participant_id) %>%
   mutate(A_row_error = lead(message, default = last(message)) == 
        'ERROR' & as.logical(first_A_row)) %>%
   ungroup

-输出

# A tibble: 10 × 6
   participant_id timestamp event first_A_row message  A_row_error
   <chr>              <dbl> <chr> <chr>       <chr>    <lgl>      
 1 ps1                 0.01 A     TRUE        ACCEPTED TRUE       
 2 ps1                 0.02 A     FALSE       ERROR    FALSE      
 3 ps1                 0.03 A     FALSE       DECLINED FALSE      
 4 ps1                 0.04 B     FALSE       ACCEPTED FALSE      
 5 ps2                 0.01 B     FALSE       HELLO    FALSE      
 6 ps2                 0.02 A     TRUE        BYE      FALSE      
 7 ps3                 0.01 A     TRUE        ACCEPTED FALSE      
 8 ps3                 0.02 A     FALSE       BYE      FALSE      
 9 ps3                 0.03 B     FALSE       ERROR    FALSE      
10 ps3                 0.04 A     FALSE       ACCEPTED FALSE