代码之家 › 专栏 › 技术社区 › hannes101

从事件指示器变量计算到事件的时间

data.table r

hannes101 · 技术社区 · 7 年前

library(data.table)
dt.MWE <- structure(list(Year = 2000:2015, Event = c(0L, 0L, 0L, 0L, 1L, 
                                       0L, 0L, 0L, 0L, 0L,
                                       1L, 0L, 0L, 0L, 0L, 
                                       0L)), row.names = c(NA, -16L)
                            , class = c("data.table", "data.frame")
                             ,  index = structure(integer(0), "`__Year`" = integer(0)))

    Year Event
 1: 2000     0
 2: 2001     0
 3: 2002     0
 4: 2003     0
 5: 2004     1
 6: 2005     0
 7: 2006     0
 8: 2007     0
 9: 2008     0
10: 2009     0
11: 2010     1
12: 2011     0
13: 2012     0
14: 2013     0
15: 2014     0
16: 2015     0

基于此,我想再增加一个专栏 Time.to.Event ,这是相对于事件发生的时间点的变量。所以2003年的变量是-1,2002年是-2,2001年是-3,2000年是-4。结果data.table应如下所示:

    Year Event Time.to.Event
 1: 2000     0            -4
 2: 2001     0            -3
 3: 2002     0            -2
 4: 2003     0            -1
 5: 2004     1             0
 6: 2005     0            -5
 7: 2006     0            -4
 8: 2007     0            -3
 9: 2008     0            -2
10: 2009     0            -1
11: 2010     1             0
12: 2011     0             1
13: 2012     0             2
14: 2013     0             3
15: 2014     0             4
16: 2015     0             5

2 回复 | 直到 7 年前

IceCreamToucan 7 年前

library(data.table)
setDT(dt.MWE)

dt.MWE[, Time.to.Event :=  seq(.N) - ifelse(any(Event), .N, 0L)
       , by = cumsum(Event) - Event]


#     Year Event Time.to.Event
#  1: 2000     0            -4
#  2: 2001     0            -3
#  3: 2002     0            -2
#  4: 2003     0            -1
#  5: 2004     1             0
#  6: 2005     0            -5
#  7: 2006     0            -4
#  8: 2007     0            -3
#  9: 2008     0            -2
# 10: 2009     0            -1
# 11: 2010     1             0
# 12: 2011     0             1
# 13: 2012     0             2
# 14: 2013     0             3
# 15: 2014     0             4
# 16: 2015     0             5

Jaap 7 年前

另一种选择:

dt.MWE[, Time.to.Event := (1:.N) - c(0,.N)[sum(Event) + 1]
       , by = cumsum(shift(Event, fill = 0))][]

它给出:

> dt.MWE
    Year Event Time.to.Event
 1: 2000     0            -4
 2: 2001     0            -3
 3: 2002     0            -2
 4: 2003     0            -1
 5: 2004     1             0
 6: 2005     0            -5
 7: 2006     0            -4
 8: 2007     0            -3
 9: 2008     0            -2
10: 2009     0            -1
11: 2010     1             0
12: 2011     0             1
13: 2012     0             2
14: 2013     0             3
15: 2014     0             4
16: 2015     0             5

推荐文章

Marco_CH · 根据动态相关的两个条件进行筛选

4 年前

Shubham · 如何在R中查找数据表的两个多类别列之间的差异/setdiff()

4 年前

mt1022 · 分组数据时的不同结果。以不同方式使用数字索引的表列

8 年前

shy zhan · 如何改变数据。有很多列的表?

8 年前

Brent Pease · 查找条件语句的最小值,其中条件也适用于之后的所有值

8 年前

Serhii · 标记数据中按组随机选择的N行。桌子

8 年前

T.Beil · 在数据表中选取均匀分布的样本

8 年前

89_Simple · 为每个组运行函数

8 年前

IceCreamToucan · 选择要保留/放入数据的组。桌子

8 年前

Valeria Lobos Ossandón · 使用R查找数据帧中其他数据帧的元素位置

8 年前