从给定行中，如何 Select R中的前几行

发布于12月17日

我有一个伪变量，如下所示:

df <- data.frame(year = seq(1990, 1997, 1),
                 x = c(1, 0, 0, 0, 1, 1, 0, 0))

year  x
1990  1
1991  0
1992  0
1993  0
1994  1
1995  1
1996  0
1997  0

如果前三年中任一年的值为非零，我想创建一个等于1的伪y.预期结果:

year  x   y
1990  1  NA
1991  0  NA
1992  0   1
1993  0   0
1994  1   1
1995  1   1
1996  0   1
1997  0   1

我该怎么做呢？最好使用dplyr%的解决方案.

推荐答案

如果您确定需要3个值，则可以执行以下操作:

library(dplyr)

df %>% mutate(y = sign((x > 0) + (lag(x) > 0) + (lag(x, 2) > 0)))
#>   year x  y
#> 1 1990 1 NA
#> 2 1991 0 NA
#> 3 1992 0  1
#> 4 1993 0  0
#> 5 1994 1  1
#> 6 1995 1  1
#> 7 1996 0  1
#> 8 1997 0  1

但如果您想 Select n，则更一般的解决方案是:

n <- 3

df %>% mutate(y = sign(purrr::reduce(seq(n) - 1, ~ .x + (lag(x, .y)))))
#>   year x  y
#> 1 1990 1 NA
#> 2 1991 0 NA
#> 3 1992 0  1
#> 4 1993 0  0
#> 5 1994 1  1
#> 6 1995 1  1
#> 7 1996 0  1
#> 8 1997 0  1