计算顺序发生的事件

Question

在我的设置中，每个城市每月都可能观察到事件的发生与否。 我想计算每个城市中连续（依次）发生了多少此类事件。

我设法实现了一个脚本，其中每个脚本内部有2个for循环，但是我有1500多个城市，大约250个月，因此循环花了很长时间。 因此，我正在寻找更快，更优雅的解决方案。

df <- data.frame(city = c(rep("a",10),rep("b",10)), month = c(1:10,1:10), event = c(0,0,0,1,0,1,1,1,0,1,1,1,0,0,0,1,1,1,1,0),                 desirable_output= c(0,0,0,1,0,1,2,3,0,1,1,2,0,0,0,1,2,3,4,0))

在此玩具示例中，“ desirable_output”列显示了我的预期输出。

预先感谢您的任何想法。

Answer 1

尝试：

library(data.table)

setDT(df)[, desirable_output := cumsum(event), by = .(city, rleid(city, event))]