R use column name string as a function parameter to reference column

Question

I've been trying to calculate the mean of a column in a data.frame inside a function which I pass teh data.frame and the name of the column to calculate but I'm not able to do so

This is an example of the data

aleatorio<-rnorm(1:52)
cod_prov<-c(1:52)
datosprov<-cbind(cod_prov, aleatorio)
rm(aleatorio,cod_prov)

This is the function

prueba<-function(bbdd,varmap){
  forMap<-noquote(paste0(substitute(bbdd),"$",substitute(varmap)))
  mean(forMap)
}

 
prueba(datosprov,"positivo")

TimTeaFan · Accepted Answer · 2021-01-08 12:16:59Z

2

This should work for data.frames. As @jay.sf points correctly out in the comments, @Otto Kässis approach is safer since it works on data.frames and matrices. The approach below is useful if you want to use bare names instead of strings to select variables in a data.frame.

prueba <- function(bbdd,varmap){
  mean(bbdd[[substitute(varmap)]])
}

prueba(iris, Sepal.Length)
#> [1] 5.843333

^{Created on 2021-01-08 by the reprex package (v0.3.0)}

edited Jan 8, 2021 at 12:16

answered Jan 8, 2021 at 11:56

TimTeaFan

18.7k4 gold badges25 silver badges51 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Otto Kässi · Accepted Answer · 2021-01-08 12:03:00Z

1

You can access columns within a dataframe by df[,'colname'], i.e.

 prueba<-function(bbdd,varmap){
   forMap<-bbdd[,varmap]
   mean(forMap)
 }

>  prueba(datosprov,"aleatorio")
[1] -0.1167871

answered Jan 8, 2021 at 12:03

Otto Kässi

3,0921 gold badge15 silver badges30 bronze badges

Collectives™ on Stack Overflow

R use column name string as a function parameter to reference column

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related