根据另一列的先前观察值创建新变量

Question

我正在构造一个新变量，该变量的值取决于另一列中的上一行。 因此，数据的顺序很重要。 这是我当前数据的外观

ID  Cong  Comm    Y
 1   52    3      0
 1   53    3      0
 1   54    3      1
 1   53    4      1
 2   50    2      1
 2   50    7      1
 3   48    4      1
 4   48    3      1
 4   48    7      0
 4   49    7      1

我想创建一个名为Y2的新变量。 如果观测值的Y = 0，则同一观测值中的Y2应等于1。如果下一行的Y = 0，则将上一个Y2值加1（此观测值的Y2值应等于2）。 继续此过程，直到Y = 1，加1，然后停止该过程。 本质上，新变量递增计数，直到另一列的值等于“ 1”并重复该过程。

它应该是这样的：

ID  Cong  Comm    Y   Y2
 1   52    3      0   1
 1   53    3      0   2
 1   54    3      1   3 
 1   53    4      1   1
 2   50    2      1   1
 2   50    7      1   1
 3   48    4      1   1
 4   48    3      1   1
 4   48    7      0   1 
 4   49    7      1   2

这是我的示例数据框。

data.frame(
ID = c(1L, 1L, 1L, 1L, 2L, 2L, 3L, 4L, 4L, 4L), 
Cong = c(52L, 53L, 54L, 53L, 50L, 50L, 48L, 48L, 48L, 49L), 
Comm = c(3L, 3L, 3L, 4L, 2L, 7L, 4L, 3L, 7L, 7L),
Y=c(0L, 0L, 1L, 1L, 1L, 1L, 1L, 1L, 0L, 1L))

循环或if-else命令是否是解决此问题的最佳方法？ 我尝试了if-else语句，但是我的代码无法正常工作。 任何建议都会很棒。

Answer 1

假设您的data.frame是df ，您可以这样做：

y = df$Y
bool=y==c(0, head(y, -1))
y[which(bool %in% F)] = 0

df$Y2 = ifelse(y==0, f7(!y), 1)

#   ID Cong Comm Y Y2
#1   1   52    3 0  1
#2   1   53    3 0  2
#3   1   54    3 1  3
#4   1   53    4 1  1
#5   2   50    2 1  1
#6   2   50    7 1  1
#7   3   48    4 1  1
#8   4   48    3 1  1
#9   4   48    7 0  1
#10  4   49    7 1  2

技巧是通过以下方式完成的：

f7 <- function(x){ tmp<-cumsum(x);tmp-cummax((!x)*tmp)}

完全定义在这篇很棒的文章中：

计算多少个连续值是正确的

最后，此解决方案是完全矢量化的，没有循环。