Создайте столбец с условными значениями, которые отражают максимальное значение в третьем столбце.

Я пытаюсь добавить столбец в датафрейм на основе сложного условия, и не могу найти посты, которые соответствуют тому, что я пытаюсь сделать. У меня есть данные следующего типа с тремя категориальными столбцами (“trial”, “word”, “point”) и одним непрерывным столбцом (“value”):

    trial  word  point value
1  trial1 word1 point1   337
2  trial1 word1 point2   105
3  trial1 word1 point3   289
4  trial1 word1 point4   190
5  trial1 word2 point1   167
6  trial1 word2 point2   223
7  trial1 word2 point3   384
8  trial1 word2 point4   143
9  trial2 word1 point1   179
10 trial2 word1 point2   101
11 trial2 word1 point3   122
12 trial2 word1 point4   213
13 trial2 word3 point1   409
14 trial2 word3 point2   221
15 trial2 word3 point3   582
16 trial2 word3 point4   178

Что я хочу знать, так это для данного trial, для данного word, какой point соответствует наивысшему value? А затем я хочу скопировать эту метку point в новый столбец для каждой строки для этого word. Вот как должен выглядеть новый столбец (max_point):

    trial  word  point value max_point
1  trial1 word1 point1   337    point1
2  trial1 word1 point2   105    point1
3  trial1 word1 point3   289    point1
4  trial1 word1 point4   190    point1
5  trial1 word2 point1   167    point3
6  trial1 word2 point2   223    point3
7  trial1 word2 point3   384    point3
8  trial1 word2 point4   143    point3
9  trial2 word1 point1   179    point4
10 trial2 word1 point2   101    point4
11 trial2 word1 point3   122    point4
12 trial2 word1 point4   213    point4
13 trial2 word3 point1   409    point3
14 trial2 word3 point2   221    point3
15 trial2 word3 point3   582    point3
16 trial2 word3 point4   178    point3

Итак, чтобы переформулировать, max_point представляет, какой point имеет наивысшее value для данного word в данном trial. Например, в trial1 word1 (первые 4 строки) из 4 точек, point1 имеет наивысшее значение (337), поэтому max_point имеет метку “point1” для каждой из 4 строк для этого слова.

Некоторые посты спрашивают, как вывести максимальное значение в новый столбец на основе условия, но не как вывести категориальную метку соответствующую максимальному значению. Как я могу это сделать?

Пожалуйста, дайте мне знать, если я могу предложить какое-либо пояснение, и спасибо!

С помощью data.table:

library(data.table)

dt[,max_point := point[which.max(value)], .(trial, word)][]
#>      trial   word  point value max_point
#>     <char> <char> <char> <int>    <char>
#>  1: trial1  word1 point1   337    point1
#>  2: trial1  word1 point2   105    point1
#>  3: trial1  word1 point3   289    point1
#>  4: trial1  word1 point4   190    point1
#>  5: trial1  word2 point1   167    point3
#>  6: trial1  word2 point2   223    point3
#>  7: trial1  word2 point3   384    point3
#>  8: trial1  word2 point4   143    point3
#>  9: trial2  word1 point1   179    point4
#> 10: trial2  word1 point2   101    point4
#> 11: trial2  word1 point3   122    point4
#> 12: trial2  word1 point4   213    point4
#> 13: trial2  word3 point1   409    point3
#> 14: trial2  word3 point2   221    point3
#> 15: trial2  word3 point3   582    point3
#> 16: trial2  word3 point4   178    point3

Данные:

dt <- fread(input = "id    trial  word  point value
            1  trial1 word1 point1   337
            2  trial1 word1 point2   105
            3  trial1 word1 point3   289
            4  trial1 word1 point4   190
            5  trial1 word2 point1   167
            6  trial1 word2 point2   223
            7  trial1 word2 point3   384
            8  trial1 word2 point4   143
            9  trial2 word1 point1   179
            10 trial2 word1 point2   101
            11 trial2 word1 point3   122
            12 trial2 word1 point4   213
            13 trial2 word3 point1   409
            14 trial2 word3 point2   221
            15 trial2 word3 point3   582
            16 trial2 word3 point4   178")[,id := NULL]

С помощью dplyr

library(dplyr)

df |>
  mutate(max_point = point[which.max(value)],
         .by = c(trial, word))
#> # A tibble: 16 × 5
#>    trial  word  point  value max_point
#>    <chr>  <chr> <chr>  <int> <chr>    
#>  1 trial1 word1 point1   337 point1   
#>  2 trial1 word1 point2   105 point1   
#>  3 trial1 word1 point3   289 point1   
#>  4 trial1 word1 point4   190 point1   
#>  5 trial1 word2 point1   167 point3   
#>  6 trial1 word2 point2   223 point3   
#>  7 trial1 word2 point3   384 point3   
#>  8 trial1 word2 point4   143 point3   
#>  9 trial2 word1 point1   179 point4   
#> 10 trial2 word1 point2   101 point4   
#> 11 trial2 word1 point3   122 point4   
#> 12 trial2 word1 point4   213 point4   
#> 13 trial2 word3 point1   409 point3   
#> 14 trial2 word3 point2   221 point3   
#> 15 trial2 word3 point3   582 point3   
#> 16 trial2 word3 point4   178 point3

Данные

df <- structure(list(trial = c("trial1", "trial1", "trial1", "trial1", 
                               "trial1", "trial1", "trial1", "trial1", "trial2", "trial2", "trial2", 
                               "trial2", "trial2", "trial2", "trial2", "trial2"), word = c("word1", 
                                                                                           "word1", "word1", "word1", "word2", "word2", "word2", "word2", 
                                                                                           "word1", "word1", "word1", "word1", "word3", "word3", "word3", 
                                                                                           "word3"), point = c("point1", "point2", "point3", "point4", "point1", 
                                                                                                               "point2", "point3", "point4", "point1", "point2", "point3", "point4", 
                                                                                                               "point1", "point2", "point3", "point4"), value = c(337L, 105L, 
                                                                                                                                                                  289L, 190L, 167L, 223L, 384L, 143L, 179L, 101L, 122L, 213L, 409L, 
                                                                                                                                                                  221L, 582L, 178L)), class = c("tbl_df", "tbl", "data.frame"), row.names = c(NA, 
                                                                                                                                                                                                                                              -16L))