R 特定比较与不同 case 数量之间的滞后差异

发布于08月29日

我需要一个专栏与"滞后的差异"之间的具体比较.在数据 struct 中，我有一些大小写重复，我不能只指定LAG=1.我还需要填充了所有大小写并带有不同比较方向(符号)的输出.

看看它的 struct :

df <- structure(
  list(
    id = c(1, 1, 2, 2, 3, 3, 3, 4, 4, 4),
    H_A = c("H",
            "A", "H", "A", "H", "H", "A", "H", "A", "A"),
    Rk = c(6, 15, 19,
           7, 8, 8, 10, 12, 3, 3)
  ),
  row.names = c(NA,-10L),
  class = c("tbl_df",
            "tbl", "data.frame")
)

我需要以下输出:

我能得到的最接近的输出是使用:

df %>% 
  group_by(id) %>% 
  mutate(Rk_diff = Rk-Rk[match('A', H_A)]) %>%
  ungroup

推荐答案

我们可以在下一步打replace个0

library(dplyr)
df %>%
   group_by(id) %>%
    mutate(Rk_diff = Rk - Rk[match('A', H_A)],
    Rk_diff = replace(Rk_diff, Rk_diff == 0, -1 * (Rk_diff[H_A != 'A'][1]))) %>% 
   ungroup

-输出

# A tibble: 10 × 4
      id H_A      Rk Rk_diff
   <dbl> <chr> <dbl>   <dbl>
 1     1 H         6      -9
 2     1 A        15       9
 3     2 H        19      12
 4     2 A         7     -12
 5     3 H         8      -2
 6     3 H         8      -2
 7     3 A        10       2
 8     4 H        12       9
 9     4 A         3      -9
10     4 A         3      -9

或者另一个选项是coalesce

library(tidyr)
df %>%
   group_by(id) %>%
   mutate(Rk_diff = case_when(H_A != 'A' ~ Rk - Rk[match('A', H_A)]), 
    Rk_diff2 = -1 * Rk_diff) %>%
    fill(Rk_diff2) %>% 
    ungroup %>% 
    mutate(Rk_diff = coalesce(Rk_diff, Rk_diff2), Rk_diff2 = NULL)

-输出

# A tibble: 10 × 4
      id H_A      Rk Rk_diff
   <dbl> <chr> <dbl>   <dbl>
 1     1 H         6      -9
 2     1 A        15       9
 3     2 H        19      12
 4     2 A         7     -12
 5     3 H         8      -2
 6     3 H         8      -2
 7     3 A        10       2
 8     4 H        12       9
 9     4 A         3      -9
10     4 A         3      -9