Как я могу условно удалить значения данных на следующий день после выполнения условия в R-кадре данных?

Question

То, с чем я работаю, это некоторые данные об окружающей среде, которые включают суточные суммы осадков. Я хочу удалить все «значения» (помеченные как «NA») для всех дней, в которых общее количество дождя превышает ноль (дождь> 0), и для всех дней после события дождя. Например, если дождь> 0 1/3/2018, все данные в столбце «value» помечены как «NA», что соответствует 1/3/2018 и 1/4 / 2018. Делать условия для того же дня достаточно просто, но я ломаю голову над тем, как ссылаться на следующий день.

Вот примерный кадр данных:

date <- seq(as.Date("2018-1-1"), as.Date("2018-1-5"), by = 
              "1 days")
plot <- rep(c(1:4), each = 5)
treatment <- rep(c('control','treat'),each = 10)
rain <- c(0,0,2,0,0,0,0,2,0,0,0,0,2,0,0,0,0,2,0,0)
value <- (seq(1:20))
df <- data.frame(date, plot, treatment, rain,value)
df$date <- rep(date, length=nrow(df))
df



date plot treatment rain value
1  2018-01-01    1   control    0     1
2  2018-01-02    1   control    0     2
3  2018-01-03    1   control    2     3
4  2018-01-04    1   control    0     4
5  2018-01-05    1   control    0     5
6  2018-01-01    2   control    0     6
7  2018-01-02    2   control    0     7
8  2018-01-03    2   control    2     8
9  2018-01-04    2   control    0     9
10 2018-01-05    2   control    0    10
11 2018-01-01    3     treat    0    11
12 2018-01-02    3     treat    0    12
13 2018-01-03    3     treat    2    13
14 2018-01-04    3     treat    0    14
15 2018-01-05    3     treat    0    15
16 2018-01-01    4     treat    0    16
17 2018-01-02    4     treat    0    17
18 2018-01-03    4     treat    2    18
19 2018-01-04    4     treat    0    19
20 2018-01-05    4     treat    0    20

Желаемый результат:

            date plot treatment rain value
1  2018-01-01    1   control    0     1
2  2018-01-02    1   control    0     2
3  2018-01-03    1   control    2    NA
4  2018-01-04    1   control    0    NA
5  2018-01-05    1   control    0     5
6  2018-01-01    2   control    0     6
7  2018-01-02    2   control    0     7
8  2018-01-03    2   control    2    NA
9  2018-01-04    2   control    0    NA
10 2018-01-05    2   control    0    10
11 2018-01-01    3     treat    0    11
12 2018-01-02    3     treat    0    12
13 2018-01-03    3     treat    2    NA
14 2018-01-04    3     treat    0    NA
15 2018-01-05    3     treat    0    15
16 2018-01-01    4     treat    0    16
17 2018-01-02    4     treat    0    17
18 2018-01-03    4     treat    2    NA
19 2018-01-04    4     treat    0    NA
20 2018-01-05    4     treat    0    20

Answer 1

(заранее: все эти решения основаны на том, что порядок задан заранее. Нетрудно выполнить внешнюю сортировку, возможно, с dplyr::arrange или с базой order.)

с dplyr это может быть сделано с:

library(dplyr)
df %>%
  group_by(plot, treatment) %>%
  mutate(value = if_else(rain > 0 | lag(rain > 0, default = FALSE), NA_integer_, value)) %>%
  ungroup()
# # A tibble: 20 x 5
#    date        plot treatment  rain value
#    <chr>      <int> <chr>     <int> <int>
#  1 2018-01-01     1 control       0     1
#  2 2018-01-02     1 control       0     2
#  3 2018-01-03     1 control       2    NA
#  4 2018-01-04     1 control       0    NA
#  5 2018-01-05     1 control       0     5
#  6 2018-01-01     2 control       0     6
#  7 2018-01-02     2 control       0     7
#  8 2018-01-03     2 control       2    NA
#  9 2018-01-04     2 control       0    NA
# 10 2018-01-05     2 control       0    10
# 11 2018-01-01     3 treat         0    11
# 12 2018-01-02     3 treat         0    12
# 13 2018-01-03     3 treat         2    NA
# 14 2018-01-04     3 treat         0    NA
# 15 2018-01-05     3 treat         0    15
# 16 2018-01-01     4 treat         0    16
# 17 2018-01-02     4 treat         0    17
# 18 2018-01-03     4 treat         2    NA
# 19 2018-01-04     4 treat         0    NA
# 20 2018-01-05     4 treat         0    20

В качестве альтернативы, с data.table

library(data.table)
DT <- as.data.table(df)
DT[rain > 0 | shift(rain > 0), value := NA, by = .(plot, treatment)]

И, возможно, менее элегантно в base-R:

do.call(rbind.data.frame,
        by(df, df[,c("plot", "treatment")], function(x) {
          n <- nrow(x)
          within(x, { value[ rain > 0 | c(FALSE, rain[-n] > 0) ] = NA })
        })
)

( хотя этот последний может не сохранять порядок строк одинаковым).

---

Данные:

df <- read.table(header=TRUE, stringsAsFactors=FALSE, text="
date plot treatment rain value
2018-01-01    1   control    0     1
2018-01-02    1   control    0     2
2018-01-03    1   control    2     3
2018-01-04    1   control    0     4
2018-01-05    1   control    0     5
2018-01-01    2   control    0     6
2018-01-02    2   control    0     7
2018-01-03    2   control    2     8
2018-01-04    2   control    0     9
2018-01-05    2   control    0    10
2018-01-01    3     treat    0    11
2018-01-02    3     treat    0    12
2018-01-03    3     treat    2    13
2018-01-04    3     treat    0    14
2018-01-05    3     treat    0    15
2018-01-01    4     treat    0    16
2018-01-02    4     treat    0    17
2018-01-03    4     treat    2    18
2018-01-04    4     treat    0    19
2018-01-05    4     treat    0    20")