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

Question

Я пытаюсь вычислить новый столбец с пользовательской функцией, которой нужны данные из той же строки и фиксированное значение, действительное для всех строк:

myfunc <- function(ds,colname,val1,col1,col2){
  # content of new column <colname> should be computed from:
  ds[colname] = val1 + ds[col1] * ds[col2] #   for each row of ds
  return(ds)
}

v1 = 2
data(mtcars) 
mt = head(mtcars) 
mt
                   mpg cyl disp  hp drat    wt  qsec vs am gear 

carb
Mazda RX4         21.0   6  160 110 3.90 2.620 16.46  0  1    4    4
Mazda RX4 Wag     21.0   6  160 110 3.90 2.875 17.02  0  1    4    4
Datsun 710        22.8   4  108  93 3.85 2.320 18.61  1  1    4    1
Hornet 4 Drive    21.4   6  258 110 3.08 3.215 19.44  1  0    3    1
Hornet Sportabout 18.7   8  360 175 3.15 3.440 17.02  0  0    3    2
Valiant           18.1   6  225 105 2.76 3.460 20.22  1  0    3    1

apply(mt,'newcol',v1,mt$wt,mt$qsec)
mt

То, что я хотел бы видеть в mt $ newcol в первой строке: 2 + 2,620 * 16,46 (-> 45,12) и все остальные строки похожи.

Итак, как я могу отправить фиксированное значение (v1) и два значения из каждой строки в мою функцию и сохранить возвращенное значение в этой строке в новом столбце?

Спасибо

Answer 1

dplyr подход:

library(dplyr)

data(mtcars) 

myfunc <- function(ds, new_column, val1, col1, col2){

  name <- rownames(ds)
  ds <- ds %>% 
    mutate(!!as.name(new_column) := val1 + !!as.name(col1) + !!as.name(col2),
           car_name = name) %>% 
    select(car_name, mpg:!!as.name(new_column))

  return(ds)

}

head(
  myfunc(ds = mtcars,
         new_column = "new_column",
         val1 = 2, 
         col1 = "hp", 
         col2 = "vs")
)

выход

           car_name  mpg cyl disp  hp drat    wt  qsec vs am gear carb new_column
1         Mazda RX4 21.0   6  160 110 3.90 2.620 16.46  0  1    4    4        112
2     Mazda RX4 Wag 21.0   6  160 110 3.90 2.875 17.02  0  1    4    4        112
3        Datsun 710 22.8   4  108  93 3.85 2.320 18.61  1  1    4    1         96
4    Hornet 4 Drive 21.4   6  258 110 3.08 3.215 19.44  1  0    3    1        113
5 Hornet Sportabout 18.7   8  360 175 3.15 3.440 17.02  0  0    3    2        177
6           Valiant 18.1   6  225 105 2.76 3.460 20.22  1  0    3    1        108