将列作为函数参数传递[重复]

Question

这个问题已经在这里有了答案：

• 如何在R的data.table中使用变量中的列名完全通用地工作 3个答案

我的df看起来像这样：

library(data.table)
df <- data.table(id = c(1,1,1,2,2), 
             starts = c(0,0,6,0,9), 
             ends = c(0,6,10,9,20))

我想创建一个函数，该函数用于创建包含平均值的新列的数据表：

mean <- function(df, col_name){
library(data.table)

a <- data.table(

mean_df = df[col_name != 0, mean(col_name)]

)
return(a)
}

然后当我运行该功能

mean(df, "starts")

这给了我错误：

 Error in data.table(mean = df[col_name != 0, mean(col_name)]) : 
    argument "col_name" is missing, with no default 

我不确定自己在做什么错-我尝试了get（）和replace（）来传递列名，但不确定是否正确执行了操作。

任何帮助表示赞赏！

EDIT1：要在函数中创建的列的已编辑名称

Answer 1

我认为像这样使用get()是非常标准的：

my_mean <- function(data, col_name) {
  data[get(col_name) != 0, mean(get(col_name))]
} 

my_mean(data = df, col_name = "starts")
[1] 7.5

BTW。 将新函数命名为mean()是一个非常糟糕的主意，因为它将替换基本函数。