r - 每次满足条件后添加 +1 (>1)

Question

(对不起，真的不知道如何更好地表达这个问题)

我有一列“拥有”，其中包含 1 和 0。我想创建一个新列“want”，每次出现 1 时，0 的值都会增加到 2，然后是 3，然后是 4，依此类推。0 永远不应该是 1。

例子:

data <- data.frame(have = (c('0', '0', '0', '0', '1', '0', '1', '0', '0', '1')), 
                   want = (c('0', '0', '0', '0', '1', '2', '1', '3', '3', '1')))                       

Answer 1

你可以这样做:

data <- data.frame(have = (c('0', '0', '0', '0', '1', '0', '1', '0', '0', '1')))  

data$want <- cumsum(as.numeric(data$have)) + 1
data$want[data$want == "1"] <- "0"
data$want[data$have == "1"] <- "1"

data
#>    have want
#> 1     0    0
#> 2     0    0
#> 3     0    0
#> 4     0    0
#> 5     1    1
#> 6     0    2
#> 7     1    1
#> 8     0    3
#> 9     0    3
#> 10    1    1

^{由 reprex package 创建于 2022-04-06| (v2.0.1)}