为R中分类变量的连续观察添加计数器？

Question

我有一个数据框，其中一个变量将我的观察结果 (ID) 和另一个分类变量 (类型) 分组。我在 R 中工作，并试图创建一个新变量来计算 ID 中相同类型的连续观察值（数据是时间序列）。请参阅下面的示例表 - Counter 是我要创建的变量。是否计算 0 并不重要。

dat <- data.frame(id = c(rep("a", 7), rep("b", 4)),
                  type = c(0, 1, 1, 2, 2, 0, 1, 1, 1, 2, 0),
                  counter = c(0, 1, 2, 1, 2, 0, 1, 1, 2, 1, 0))

到目前为止，我能够制作计数器，虽然可能不是以最有效的方式，但我正在努力在分组 (ID) 中执行此操作并有效地在 type = 1 和 type = 2 之间切换。我编写的代码使计数器在下面。关于如何有效地做到这一点的任何想法？谢谢。

dat$counter <- 0
counter     <- 0
for(i in 1:nrow(dat)){
  if(dat[i,"type"] != 0){
    counter <- counter + 1
    dat[i,"count"] <- counter
    # use to differentiate between 1 and 2?
    this_group <- dat[i,"type"]
  }
  if(dat[i,"type"] == 0){
    counter <- 0
  }
}

Answer 1

对于每个id、type 和以type 0 开头的连续行，使用ave 和seq_along 创建一个序列。没有使用任何包：

transform(dat, 
  counter = (type > 0) * ave(type, id, type, cumsum(type == 0), FUN = seq_along))

给予：

   id type counter
1   a    0       0
2   a    1       1
3   a    1       2
4   a    2       1
5   a    2       2
6   a    0       0
7   a    1       1
8   b    1       1
9   b    1       2
10  b    2       1
11  b    0       0

Answer 2

library(dplyr)

dat %>%
  mutate(counter = case_when(
    type == 0 ~ 0,
    id != lag(id) ~ 1,
    type != lag(type) ~ 1
  )) %>%
  mutate(counter = case_when(
    id != lag(id) ~ 1,
    type == lag(type) ~ lag(counter) + 1,
    T ~ counter
  ))

结果：

   id type counter
1   a    0       0
2   a    1       1
3   a    1       2
4   a    2       1
5   a    2       2
6   a    0       0
7   a    1       1
8   b    1       1
9   b    1       2
10  b    2       1
11  b    0       0

注意：一些代码在两个case_when() 参数中重复，但这些重复对于涵盖许多条件是必要的。

我在各种条件下测试了这段代码，它看起来很健壮。它无法处理的一件事是如果您在最后一个值处重复（它将返回NA，因为我使用lag()）。

这段较长的代码也涵盖了这种情况，并且（我认为）无论您的代码如何都应该可以工作：

dat %>%
  mutate(counter = case_when(
    type == 0 ~ 0,
    type != lag(type) ~ 1
  )) %>%
  mutate(counter = case_when(
    id != lag(id) ~ 1,
    type == lag(type) ~ lag(counter) + 1,
    T ~ counter
  )) %>%
  mutate(counter = case_when(
    is.na(counter) ~ lag(counter) + 1,
    T ~ counter
  ))