Utilisation dataframe Nom de la colonne comme R Fonction Argument

Question

fun <- function(dataframe, var){ 
    print(data$var) 
} 

dataset <- data.frame(a = 1:6, b = 12:17, c = 3:8) 

fun(dataset, a) 

Essentiellement, je voudrais faire quelque chose comme ça où je peux passer l'un des noms de colonnes en fonction et être capable de l'utiliser pour appeler cette colonne plus tard. Je sais que la fonction group_by le gère d'une manière ou d'une autre, mais quand je suis passé en mode de débogage et que j'ai essayé de le comprendre, je ne pouvais pas faire la tête ou la queue de ce qui était en train de faire ce que je voulais.

Answer 1

Pour ce faire dplyr-like, vous devez utiliser tidyevaluation. Tout d'abord, vous devez enquo()var, puis reportez-vous à var avec !!:

library(dplyr) 

fun <- function(dataframe, var){ 
    var <- enquo(var) 
    select(dataframe, !!var) 
} 

dataset <- data.frame(a = 1:6, b = 12:17, c = 3:8) 

fun(dataset, a) 

Vous pouvez trouver un bon tuto tidyevalhere.

Best,

Colin