использование dplyr для расчета дней подряд с определенным значением - PullRequest
Новая
       11

использование dplyr для расчета дней подряд с определенным значением

2 голосов
/ 13 мая 2019

Я пытаюсь рассчитать индекс тяжести зимы, и один из компонентов индекса требует вычисления последовательного числа дней <0 градусов по Цельсию до и с учетом этой конкретной даты.Например: </p> 

Day 1 = 2 degrees C  
Day 2 = -2 degrees C  
Day 3 = -5 degrees C

Таким образом, значение, которое я пытаюсь вычислить (называемое временными числами), равно 0 для первого дня;1 день 2;и 2 для Дня 3.

Вот пример, показывающий, как выглядят данные: 

dat <- tibble(
  date = seq(as.Date('2010-01-01'), as.Date('2010-01-10'), 1),
  temp = c(4.2, 3.35, -0.6, -0.25, 0.8, 0.8, -2.5, -5.25, -0.5, 3.35)
)
dat

 date        temp
   <date>     <dbl>
 1 2010-01-01  4.2 
 2 2010-01-02  3.35
 3 2010-01-03 -0.6 
 4 2010-01-04 -0.25
 5 2010-01-05  0.8 
 6 2010-01-06  0.8 
 7 2010-01-07 -2.5 
 8 2010-01-08 -5.25
 9 2010-01-09 -0.5 
10 2010-01-10  3.35

Вот еще один набор данных, начинающийся со значения временного значения меньше нуля, так как это, казалось, вызывало проблему: 

dat2 <- tibble(
  date = seq(as.Date('2010-01-01'), as.Date('2010-01-10'), 1),
  temp = c(-1.95, -1.1, -2.8, -6.7, 1.4, 4.45, 6.1, 4.7, -1.7, -3.9)
)
dat2

Итак, dat2 должен выглядеть так: 

date        temp tempdays
   <date>     <dbl>    <dbl>
 1 2010-01-01 -1.95        1
 2 2010-01-02 -1.1         2
 3 2010-01-03 -2.8         3
 4 2010-01-04 -6.7         4
 5 2010-01-05  1.4         0
 6 2010-01-06  4.45        0
 7 2010-01-07  6.1         0
 8 2010-01-08  4.7         0
 9 2010-01-09 -1.7         1
10 2010-01-10 -3.9         2

Я думаю, lag() можно использовать для этого?

Ответы [ 2 ]

3 голосов
/ 13 мая 2019

Вы можете создать группирующую переменную, используя cumsum, а затем использовать row_number для генерации последовательных дней, когда температура была меньше 0. 

library(dplyr)

dat %>%
  group_by(group = cumsum(temp > 0)) %>%
  mutate(tempdays = row_number() - 1) %>%
  ungroup() %>%
  select(-group)


#    date      temp tempdays
#   <date>     <dbl>    <dbl>
# 1 2010-01-01  4.2         0
# 2 2010-01-02  3.35        0
# 3 2010-01-03 -0.6         1
# 4 2010-01-04 -0.25        2
# 5 2010-01-05  0.8         0
# 6 2010-01-06  0.8         0
# 7 2010-01-07 -2.5         1
# 8 2010-01-08 -5.25        2
# 9 2010-01-09 -0.5         3
#10 2010-01-10  3.35        0

и с использованием базы R, которая будет с ave

with(dat, ave(temp, cumsum(temp > 0), FUN = seq_along) - 1)

РЕДАКТИРОВАТЬ

Это не работает, как ожидалось, если первая группа отрицательна.Вот обновленная версия с использованием rle, которая работает с dat, а также dat2

dat2 %>%
  mutate(tempdays = with(rle(temp < 0), rep(values, lengths))) %>%
  group_by(group = cumsum(temp > 0)) %>%
  mutate(tempdays = cumsum(tempdays)) %>%
  ungroup() %>%
  select(-group)


#      date    temp   tempdays
#     <date>   <dbl>    <int>
# 1 2010-01-01 -1.95        1
# 2 2010-01-02 -1.1         2
# 3 2010-01-03 -2.8         3
# 4 2010-01-04 -6.7         4
# 5 2010-01-05  1.4         0
# 6 2010-01-06  4.45        0
# 7 2010-01-07  6.1         0
# 8 2010-01-08  4.7         0
# 9 2010-01-09 -1.7         1
#10 2010-01-10 -3.9         2
1 голос
/ 13 мая 2019

Мы можем использовать data.table

library(data.table)
setDT(dat)[, tempdays := seq_len(.N) -1 , cumsum(temp > 0)]
dat
#          date  temp tempdays
# 1: 2010-01-01  4.20        0
# 2: 2010-01-02  3.35        0
# 3: 2010-01-03 -0.60        1
# 4: 2010-01-04 -0.25        2
# 5: 2010-01-05  0.80        0
# 6: 2010-01-06  0.80        0
# 7: 2010-01-07 -2.50        1
# 8: 2010-01-08 -5.25        2
# 9: 2010-01-09 -0.50        3
#10: 2010-01-10  3.35        0
...