R data.table用于确定某人是新人还是现有人

时间:2016-01-06 10:11:28

标签: r data.table

我有以下data.table

year      Person     Number_of_visits
2012      1          0
2013      1          4
2014      1          0
2015      1          1
2012      2          1
2013      2          5 
...

我想确定每个人,他们的第一次访问是哪一年。所以期望的输出是:

year      Person     Number_of_visits    New?
2012      1          0                   NA
2013      1          4                   Yes
2014      1          0                   No
2015      1          1                   No
2012      2          1                   NA
2013      2          5                   No

我认为也许可以在data.table中使用SHIFT函数,但我无法弄清楚如何这样做。一旦一个人访问过,他/她就不再是新人了,即使可能有一年之后没有访问。如果第一次访问发生在2012年,则应该有NA或类似条目。

我用过

test <- DT[ , NEW := c(0, (2:1)[(Number_of_visits== shift(Number_of_visits)) + 1][-1]), by = Person]

但这自然给了我所有的变化,我想只将0的第一个变化记录到0以上的某个值(访问次数)

1 个答案:

答案 0 :(得分:7)

我会将其分解为以下步骤(我确信解决方案可以打高尔夫球到更短的时间)

setorder(dt, Person, year) # Make sure the order is correct
dt[, New := "No"] # Set No as default
dt[dt[, .I[which.max(Number_of_visits > 0)], by = Person]$V1, New := "Yes"] # find first visits
dt[year == 2012, New := NA_character_] # Set NAs to 2012
dt
#    year Person Number_of_visits New
# 1: 2012      1                0  NA
# 2: 2013      1                4 Yes
# 3: 2014      1                0  No
# 4: 2015      1                1  No
# 5: 2012      2                1  NA
# 6: 2013      2                5  No