如何在R的用户自定义函数（UDF）中自定义数据框名称

在R里没法直接通过字符串拼接变量名来赋值，你可以用以下两种方式实现需求：

方法1：在函数内直接创建全局命名数据框 #

用assign()函数可以把生成的数据框以指定名称存入全局环境，修改后的函数如下：

calculatingMean <- function(df, colname){
  ft_cate <- df[df$Satisfaction == 'Neutral or Dissatisfied',c(10:23)]
  colavg <- as.data.frame(colMeans(ft_cate))
  # 用传入的colname拼接唯一名称，比如"children_avg"
  assign(paste0(colname, "_avg"), colavg, envir = .GlobalEnv)
}

调用后，全局环境里就会生成children_avg这个数据框：

calculatingMean(children, 'children')

方法2：返回命名列表，再转为全局变量（更安全） #

如果不想直接修改全局环境，可以让函数返回一个命名列表，之后用list2env()把列表元素转为全局变量，这种方式更可控：

calculatingMean <- function(df, colname){
  ft_cate <- df[df$Satisfaction == 'Neutral or Dissatisfied',c(10:23)]
  colavg <- as.data.frame(colMeans(ft_cate))
  # 返回以colname为名称的列表
  setNames(list(colavg), paste0(colname, "_avg"))
}

# 调用并转为全局变量
result_list <- calculatingMean(children, 'children')
list2env(result_list, envir = .GlobalEnv)

为什么你之前的方法无效？ #

R里的变量名不能通过colavg + colname这种字符串拼接的方式直接定义，必须用assign()这类可以操作变量名称的函数，或者通过命名列表的方式间接生成。

内容的提问来源于stack exchange，提问作者william_Li