Question

我阅读了很多，测试了很多，但没有成功。我正在为每个组创建一个计数器，但是此提示器会在每次组出现时重新启动（即使它“重新出现”）。

示例：

我需要每个TYPE_OF_PROMOTION的seq_along，并根据组外观重新启动。结果应类似于counter 由于促销B在第18周重新出现，但又是新出现，请重新启动计数器。

如果该解决方案也适用于NAs-好的，我做了一个快速演示。

   Week TYPE_OF_PROMOTION counter
11   11              <NA>      NA
12   12                 B       1
13   13                 B       2
14   14                 B       3
15   15              <NA>      NA
16   16              <NA>      NA
17   17              <NA>      NA
18   18                 B       1
19   19                 B       2
20   20              <NA>      NA

Answer 1

这里是data.table的一个选项。将'data.frame'转换为'data.table'（setDT(df1)），并按'TYPE_OF_PROMOTION'的运行长度ID分组，创建行序列（seq_len(.N)）乘以NA在“ TYPE_OF_PROMOTION”中创建“计数器”列

library(data.table)
setDT(df1)[, counter2 := seq_len(.N) * NA^is.na(TYPE_OF_PROMOTION), 
      by = rleid(TYPE_OF_PROMOTION)]
df1
#    Week TYPE_OF_PROMOTION counter counter2
# 1:   11              <NA>      NA       NA
# 2:   12                 B       1        1
# 3:   13                 B       2        2
# 4:   14                 B       3        3
# 5:   15              <NA>      NA       NA
# 6:   16              <NA>      NA       NA
# 7:   17              <NA>      NA       NA
# 8:   18                 B       1        1
# 9:   19                 B       2        2
#10:   20              <NA>      NA       NA

或者使用dplyr中的rleid中的data.table

library(dplyr)
df1 %>%
    group_by(grp = rleid(TYPE_OF_PROMOTION)) %>%
    mutate(counter2 = case_when(is.na(TYPE_OF_PROMOTION) ~ NA_integer_, 
                              TRUE ~ row_number())) %>%
    ungroup %>%
    select(-grp)

或将base R与rle一起使用

with(df1, with(rle(!is.na(TYPE_OF_PROMOTION)), 
          sequence(lengths) * rep(NA^!values, lengths)))
#[1] NA  1  2  3 NA NA NA  1  2 NA

数据

df1 <- structure(list(Week = 11:20, TYPE_OF_PROMOTION = c(NA, "B", "B", 
"B", NA, NA, NA, "B", "B", NA), counter = c(NA, 1L, 2L, 3L, NA, 
NA, NA, 1L, 2L, NA)), class = "data.frame", row.names = c("11", 
 "12", "13", "14", "15", "16", "17", "18", "19", "20"))

R：每组出现时重启计数器

1 个答案:

数据