根据 R 中每个 ID 的 eventdate 填写订单

Question

我想问一个关于根据每个 ID 的 eventdate 填写订单的简单问题。

我有下表。 我想创建一个名为 order 的新列，以便 eventdate 不是 NA 的那一天，它的 order 为零。 然后，我想将负连续递减整数填充到前一行，将正连续递增整数填充到每个 ID 的下面行。

+----+------------+-------+
| ID | eventdate  | Value |
+----+------------+-------+
|  1 | NA         |    10 |
|  1 | NA         |    11 |
|  1 | NA         |    12 |
|  1 | NA         |    11 |
|  1 | 2011-03-18 |    15 |
|  1 | NA         |    17 |
|  1 | NA         |    18 |
|  1 | NA         |    15 |
|  2 | NA         |     5 |
|  2 | NA         |     6 |
|  2 | NA         |     7 |
|  2 | 2011-05-28 |     9 |
|  2 | NA         |    10 |
|  2 | NA         |    11 |
|  2 | NA         |    15 |
|  2 | NA         |    16 |
|  3 | NA         |    20 |
|  3 | NA         |    22 |
|  3 | NA         |    23 |
|  3 | NA         |    24 |
|  3 | 2012-05-28 |    28 |
|  3 | NA         |    29 |
|  3 | NA         |    25 |
|  3 | NA         |    24 |
|  3 | NA         |    26 |
|  3 | NA         |    24 |
+----+------------+-------+

总之，我想做下表

+----+------------+-------+-------+
| ID | eventdate  | Value | order |
+----+------------+-------+-------+
|  1 | NA         |    10 |    -4 |
|  1 | NA         |    11 |    -3 |
|  1 | NA         |    12 |    -2 |
|  1 | NA         |    11 |    -1 |
|  1 | 2011-03-18 |    15 |     0 |
|  1 | NA         |    17 |     1 |
|  1 | NA         |    18 |     2 |
|  1 | NA         |    15 |     3 |
|  2 | NA         |     5 |    -3 |
|  2 | NA         |     6 |    -2 |
|  2 | NA         |     7 |    -1 |
|  2 | 2011-05-28 |     9 |     0 |
|  2 | NA         |    10 |     1 |
|  2 | NA         |    11 |     2 |
|  2 | NA         |    15 |     3 |
|  2 | NA         |    16 |     4 |
|  3 | NA         |    20 |    -4 |
|  3 | NA         |    22 |    -3 |
|  3 | NA         |    23 |    -2 |
|  3 | NA         |    24 |    -1 |
|  3 | 2012-05-28 |    28 |     0 |
|  3 | NA         |    29 |     1 |
|  3 | NA         |    25 |     2 |
|  3 | NA         |    24 |     3 |
|  3 | NA         |    26 |     4 |
|  3 | NA         |    24 |     5 |
+----+------------+-------+-------+

非常感谢您！

Answer 1

您可以在 eventdate 中找到非 NA 值的eventdate并用该组中的行 position 减去它。

library(dplyr)

df <- df %>%
  group_by(ID) %>%
  mutate(order = row_number() - match(TRUE, !is.na(eventdate))) %>%
  ungroup 
df

# A tibble: 26 × 4
#      ID eventdate  Value order
#   <int> <chr>      <int> <int>
# 1     1 NA            10    -4
# 2     1 NA            11    -3
# 3     1 NA            12    -2
# 4     1 NA            11    -1
# 5     1 2011-03-18    15     0
# 6     1 NA            17     1
# 7     1 NA            18     2
# 8     1 NA            15     3
# 9     2 NA             5    -3
#10     2 NA             6    -2
# … with 16 more rows

在基础 R 中，同样的事情可以写成 -

df <- transform(df, order = ave(!is.na(eventdate), ID, 
              FUN = function(x) seq_along(x) - match(TRUE, x)))