时间戳

首页 > 解决方案 > R中的增量指示符列

r - R中的增量指示符列

问题描述

我有以下数据，我正在尝试创建一个指标列来跟踪 id 值的增量。

df <- data.frame(id=c(1,1,1,1,1,1,1,2,2,2,3,3,4),
         rank=c(1,2,3,1,1,1,2,2,3,1,1,2,3),dates=c('2019-06-15','2019-07-15','2019-08-15','2019-09-15','2019-10-15','2019-11-15','2019-12-15',
                 '2019-10-15','2019-11-15','2019-12-15',
                 '2019-11-15','2019-12-15','2019-12-15'),new_col=c(0,1,1,1,1,1,1,0,1,1,0,1,0)
         )

日期的顺序在这里很重要。

setDT(df)[order(id,dates),]

期望的输出

   id  rank   dates     new_col
1   1    1   2019-06-15       0
2   1    2   2019-07-15       1
3   1    3   2019-08-15       1
4   1    1   2019-09-15       1
5   1    1   2019-10-15       1
6   1    1   2019-11-15       1
7   1    2   2019-12-15       1
8   2    2   2019-10-15       0
9   2    3   2019-11-15       1
10  2    1   2019-12-15       1
11  3    1   2019-11-15       0
12  3    2   2019-12-15       1
13  4    3   2019-12-15       0

更新的输入数据-

df <- data.frame(id=c(1,1,1,1,1,1,1,2,2,2,3,3,4,5,5,5),
             rank=c(1,2,3,1,1,1,2,2,3,1,1,2,3,1,1,1),dates=c('2019-06-15','2019-07-10','2019-08-15','2019-09-15','2019-10-15','2019-11-15','2019-12-15',
                     '2019-10-15','2019-11-15','2019-12-15',
                     '2019-11-15','2019-12-15','2019-12-15','2019-10-15','2019-11-15','2019-12-15'))

更新输出-

      id   rank  dates       new

 1     1     1 2019-06-15     0
 2     1     2 2019-07-10     1
 3     1     3 2019-08-15     1
 4     1     1 2019-09-15     1
 5     1     1 2019-10-15     1
 6     1     1 2019-11-15     1
 7     1     2 2019-12-15     1
 8     2     2 2019-10-15     0
 9     2     3 2019-11-15     1
10     2     1 2019-12-15     1
11     3     1 2019-11-15     0
12     3     2 2019-12-15     1
13     4     3 2019-12-15     0
14     5     1 2019-10-15     0
15     5     1 2019-11-15     0
16     5     1 2019-12-15     0

标签： rdplyrdata.tableplyr

解决方案

我们可以 diff在按“id”分组后在“日期”上使用

library(dplyr)
df %>% 
   group_by(id) %>%       
   mutate(new = c(0, diff(as.Date(dates)) > 0))
# A tibble: 13 x 5
# Groups:   id [4]
#      id  rank dates      new_col   new
#   <dbl> <dbl> <fct>        <dbl> <dbl>
# 1     1     1 2019-06-15       0     0
# 2     1     2 2019-07-15       1     1
# 3     1     3 2019-08-15       1     1
# 4     1     1 2019-09-15       1     1
# 5     1     1 2019-10-15       1     1
# 6     1     1 2019-11-15       1     1
# 7     1     2 2019-12-15       1     1
# 8     2     2 2019-10-15       0     0
# 9     2     3 2019-11-15       1     1
#10     2     1 2019-12-15       1     1
#11     3     1 2019-11-15       0     0
#12     3     2 2019-12-15       1     1
#13     4     3 2019-12-15       0     0

更新

df %>% 
    group_by(id) %>% 
    mutate(new = +(c(FALSE, diff(as.Date(dates)) > 0) & 
       (any(rank  != lag(rank, default = first(rank))))))
# A tibble: 16 x 4
# Groups:   id [5]
#      id  rank dates        new
#   <dbl> <dbl> <fct>      <int>
# 1     1     1 2019-06-15     0
# 2     1     2 2019-07-10     1
# 3     1     3 2019-08-15     1
# 4     1     1 2019-09-15     1
# 5     1     1 2019-10-15     1
# 6     1     1 2019-11-15     1
# 7     1     2 2019-12-15     1
# 8     2     2 2019-10-15     0
# 9     2     3 2019-11-15     1
#10     2     1 2019-12-15     1
#11     3     1 2019-11-15     0
#12     3     2 2019-12-15     1
#13     4     3 2019-12-15     0
#14     5     1 2019-10-15     0
#15     5     1 2019-11-15     0
#16     5     1 2019-12-15     0

推荐阅读