创建一个列，根据另一行中的事件为数据框中的一行赋值

Question

我有一个结构如下的数据框：

example <- data.frame(id = c(1,1,1,1,1,1,1,2,2,2,2,2),
                      event = c("email","email","email","draw","email","email","draw","email","email","email","email","draw"),
                      date = c("2020-03-01","2020-06-01","2020-07-15","2020-07-28","2020-08-07","2020-09-01","2020-09-15","2020-05-22","2020-06-15","2020-07-13","2020-07-15","2020-07-31"),
                      amount = c(NA,NA,NA,10000,NA,NA,1500,NA,NA,NA,NA,2200))

这是数据框的简化版本。 我正在尝试创建一个列，该列将在绘制事件之前为最后一封电子邮件分配一个 1，以及一个将在与电子邮件相同的行上绘制的金额的列。 所需的数据框如下所示：

desiredResult <- data.frame(id = c(1,1,1,1,1,1,1,2,2,2,2,2),
                      event = c("email","email","email","draw","email","email","draw","email","email","email","email","draw"),
                      date = c("2020-03-01","2020-06-01","2020-07-15","2020-07-28","2020-08-07","2020-09-01","2020-09-15","2020-05-22","2020-06-15","2020-07-13","2020-07-15","2020-07-31"),
                      amount = c(NA,NA,NA,10000,NA,NA,1500,NA,NA,NA,NA,2200),
                      EmailBeforeDrawFlag = c(NA,NA,1,NA,NA,1,NA,NA,NA,NA,1,NA),
                      EmailBeforeDrawAmount = c(NA,NA,10000,NA,NA,1500,NA,NA,NA,NA,2200,NA))

Answer 1

这是dplyr解决方案。 当你创建新列，要使用if_else()中的定义EmailBeforeDrawFlag检验一个条件，而lead功能上一行去寻找event 。 EmailBeforeDrawAmount是突出的lead(amount) 。

example %>%
  mutate(EmailBeforeDrawFlag = if_else(lead(event) == "draw", 1, NA_real_ ),
         EmailBeforeDrawAmount = lead(amount))
   id event       date amount EmailBeforeDrawFlag EmailBeforeDrawAmount
1   1 email 2020-03-01     NA                  NA                    NA
2   1 email 2020-06-01     NA                  NA                    NA
3   1 email 2020-07-15     NA                   1                 10000
4   1  draw 2020-07-28  10000                  NA                    NA
5   1 email 2020-08-07     NA                  NA                    NA
6   1 email 2020-09-01     NA                   1                  1500
7   1  draw 2020-09-15   1500                  NA                    NA
8   2 email 2020-05-22     NA                  NA                    NA
9   2 email 2020-06-15     NA                  NA                    NA
10  2 email 2020-07-13     NA                  NA                    NA
11  2 email 2020-07-15     NA                   1                  2200
12  2  draw 2020-07-31   2200                  NA                    NA

Answer 2

我们还可以利用NA^在lead上创建列

library(dplyr)
example %>%
      mutate(EmailBeforeDrawFlag = NA^(lead(event != 'draw')), 
             EmailBeforeDrawAmount = lead(amount))

-输出

#    id event       date amount EmailBeforeDrawFlag EmailBeforeDrawAmount
#1   1 email 2020-03-01     NA                  NA                    NA
#2   1 email 2020-06-01     NA                  NA                    NA
#3   1 email 2020-07-15     NA                   1                 10000
#4   1  draw 2020-07-28  10000                  NA                    NA
#5   1 email 2020-08-07     NA                  NA                    NA
#6   1 email 2020-09-01     NA                   1                  1500
#7   1  draw 2020-09-15   1500                  NA                    NA
#8   2 email 2020-05-22     NA                  NA                    NA
#9   2 email 2020-06-15     NA                  NA                    NA
#10  2 email 2020-07-13     NA                  NA                    NA
#11  2 email 2020-07-15     NA                   1                  2200
#12  2  draw 2020-07-31   2200                  NA                    NA