如何编写R自定义函数修改数据框df中用$索引的指定列？

• 你用的<<-是父环境全局赋值运算符，函数内的形参x接收的是df$v1的数值拷贝，x<<-gsub("_","",x)实际是在全局环境创建了一个名为x的新变量，完全不会修改原数据框df的对应列，因此调用后原df没有任何变化。
• 这种逐列手动调用函数的写法效率低，不适合多列场景。

基础自定义函数写法 #

首先调整函数逻辑，不要用全局赋值，直接返回处理后的结果，需要修改哪列就把返回值赋值给对应列即可：

# 自定义去下划线函数
remove_underscore <- function(x) {
  gsub(pattern = "_", replacement = "", x = x)
}

# 单列修改直接赋值回去
df$v1 <- remove_underscore(df$v1)
df$v2 <- remove_underscore(df$v2)
df$v3 <- remove_underscore(df$v3)

批量全列处理写法 #

如果需要处理数据框所有列，直接用lapply批量应用函数即可，不需要逐列手动调用：

# 对df所有列应用去下划线函数，直接覆盖原df
df[] <- lapply(X = df, FUN = remove_underscore)

如果你的数据框包含非字符列，不想修改非字符列的话，可以给函数加类型判断：

remove_underscore <- function(x) {
  # 仅处理字符类型的列
  if (is.character(x)) {
    return(gsub("_", "", x))
  }
  # 非字符列直接返回原值
  return(x)
}

df[] <- lapply(df, remove_underscore)

用你提供的示例数据测试，运行上述代码后df的输出结果如下：

v1 v2 v3
1 a1 a2 a3
2 b1 b2 b3

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