R 计算有条件的连续天数

Question

我有下一个问题。如果我有下一个数据框

data<- data.frame(
        Time= c("2018-01-01", "2018-01-02", "2018-01-03", "2018-01-04", "2018-01-05"),
        TEN= c(10,20,11,12,16)
    )

我想使用 R< 15 as a new column 来计算 10 的连续天数。

我尝试用

waves_min <- function(df, prop, min_value, min_days) {
    sum(with(rle(df$Temp > min_temp), values & lengths >= min_days))
}

但它返回该计数的总数，而不是每行的值。

有什么想法吗？

谢谢

Answer 1

dplyr

+

data.table::rleid

:

library(dplyr)
library(data.table)

data %>%
  group_by(ID = data.table::rleid(TEN < 15)) %>%
  mutate(Consec_Days = if_else(TEN < 15, row_number(), 0L))

输出：

# A tibble: 7 x 4
# Groups:   ID [5]
  Time         TEN    ID Consec_Days
  <fct>      <dbl> <int>       <int>
1 2018-01-01    10     1           1
2 2018-01-02    20     2           0
3 2018-01-03    11     3           1
4 2018-01-04    12     3           2
5 2018-01-05    16     4           0
6 2018-01-06    17     4           0
7 2018-01-07    14     5           1

data.table

：

setDT(data)
data[, Consec_Days := ifelse(TEN < 15, 1:.N, 0L), by = rleid(TEN < 15)]

输出：

         Time TEN Consec_Days
1: 2018-01-01  10           1
2: 2018-01-02  20           0
3: 2018-01-03  11           1
4: 2018-01-04  12           2
5: 2018-01-05  16           0
6: 2018-01-06  17           0
7: 2018-01-07  14           1

基础 R +

data.table::rleid

:

data$Consec_Days <- with(data, ave(TEN, data.table::rleid(TEN < 15), 
                                   FUN = function(x) ifelse(x < 15, seq_along(x), 0L)))

输出：

        Time TEN Consec_Days
1 2018-01-01  10           1
2 2018-01-02  20           0
3 2018-01-03  11           1
4 2018-01-04  12           2
5 2018-01-05  16           0
6 2018-01-06  17           0
7 2018-01-07  14           1

数据：

data <- data.frame(
  Time= c("2018-01-01", "2018-01-02", "2018-01-03", "2018-01-04", "2018-01-05", "2018-01-06", "2018-01-07"),
  TEN= c(10,20,11,12,16,17,14)
)

我在 OP 的示例数据中添加了更多行，以说明这些解决方案适用于所有情况。基本思想是查看逻辑比较

TEN < 15

，它返回一个布尔向量。然后，

rleid

创建一个“游程长度编码 ID”向量，每次从

TRUE

到

FALSE

切换时，它都会加一，反之亦然。然后，我们按此

ID

列进行分组，以计算

TEN < 15

的连续天数。