Добавление переменной к аккуратным данным, значение которых основано на расчете строки столбца

Question

Мой текущий проект содержит данные, структура которых имеет вид:

my_df <- tibble(
  zn = c("hm","hm", "hm", "h60","h60","h60", "h85","h85","h85"),
  nm = c("c", "cA", "cB", "c","cA", "cB", "c","cA", "cB"), 
  val = c(60, 40, 20, 250, 150, 100, 400, 250, 150),
  znt = c(100, 100, 100, 300, 300, 300, 500, 500, 500), 
)

my_df
# A tibble: 9 x 4
zn    nm      val   znt
  <chr> <chr> <dbl> <dbl>
1 hm    c        60   100
2 hm    cA       40   100
3 hm    cB       20   100
4 h60   c       250   300
5 h60   cA      150   300
6 h60   cB      100   300
7 h85   c       400   500
8 h85   cA      250   500
9 h85   cB      150   500

И я хотел бы добавить новую переменную c0, значение которой определено c0 = znt - c для каждого zn. Конечный результат будет выглядеть так:

# A tibble: 12 x 4
   zn    nm      val   znt
   <chr> <chr> <dbl> <dbl>
 1 hm    c0       40   100
 2 hm    c        60   100
 3 hm    cA       40   100
 4 hm    cB       20   100
 5 h60   c0       50   300
 6 h60   c       250   300
 7 h60   cA      150   300
 8 h60   cB      100   300
 9 h85   c0      100   500
10 h85   c       400   500
11 h85   cA      250   500
12 h85   cB      150   500

У меня есть идея, как это сделать, но это кажется очень запутанным, и я надеюсь, что есть лучший способ. Если бы это можно было сделать, используя что-то из tidyverse, это тоже было бы здорово.

Answer 1

Другой вариант:

library(tidyverse)

my_df %>%
  group_split(zn) %>%
  map_dfr(~ add_row(.data = ., 
                    zn = .$zn[1], 
                    nm = 'c0', 
                    val = .$znt[1] - .$val[.$nm == 'c'], 
                    znt = .$znt[1],
                    .before = 1))

Выход:

# A tibble: 12 x 4
   zn    nm      val   znt
 * <chr> <chr> <dbl> <dbl>
 1 h60   c0       50   300
 2 h60   c       250   300
 3 h60   cA      150   300
 4 h60   cB      100   300
 5 h85   c0      100   500
 6 h85   c       400   500
 7 h85   cA      250   500
 8 h85   cB      150   500
 9 hm    c0       40   100
10 hm    c        60   100
11 hm    cA       40   100
12 hm    cB       20   100

Answer 2

my_df %>% 
   bind_rows(my_df %>% filter(nm == "c") %>% 
   mutate(nm = "c0" ,  val = znt - val)) %>% 
   arrange(zn, nm)

т

zn    nm      val   znt
   <chr> <chr> <dbl> <dbl>
 1 h60   c       250   300
 2 h60   c0       50   300
 3 h60   cA      150   300
 4 h60   cB      100   300
 5 h85   c       400   500
 6 h85   c0      100   500
 7 h85   cA      250   500
 8 h85   cB      150   500
 9 hm    c        60   100
10 hm    c0       40   100
11 hm    cA       40   100
12 hm    cB       20   100

Answer 3

Другой вариант - go в ширину, затем вычислить, затем go в длину.

my_df %>%
  pivot_wider(names_from = "nm", values_from = "val") %>%
  mutate(c0 = znt - c) %>%
  pivot_longer(cols = -c(zn, znt), names_to = "nm", values_to = "val")
#> # A tibble: 12 x 4
#>    zn      znt nm      val
#>    <chr> <dbl> <chr> <dbl>
#>  1 hm      100 c        60
#>  2 hm      100 cA       40
#>  3 hm      100 cB       20
#>  4 hm      100 c0       40
#>  5 h60     300 c       250
#>  6 h60     300 cA      150
#>  7 h60     300 cB      100
#>  8 h60     300 c0       50
#>  9 h85     500 c       400
#> 10 h85     500 cA      250
#> 11 h85     500 cB      150
#> 12 h85     500 c0      100