如何在R语言中自动重命名函数内的array/vector变量？

我写了一个可用但不够简洁的函数，作用是从数据框指定列生成长向量，每次ID变化时插入若干NA值。现在希望在函数内自动给生成的array变量重命名，比如生成array_Score1这类独特名称，避免运行不同列时变量被覆盖。

尝试了以下写法但报错，推测是字符串未被识别为变量名：

c("array_", as.character(x)) <- array
rm(array)
print(c("array_", as.character(x)))

示例数据及现有函数代码 #

ID <- c(rep ("A", 3), rep("B", 3))
Day <- c(1,2,3,1,2,3)
Score1 <- c(12,4, 16, 9, 12, 13)
Score2 <- c(1, 4, 4, 1, 3, 5)
Score3 <- c(23, 19, 12, 12, 24, 11)

df <- data.frame(ID, Day, Score1, Score2, Score3)
print(df)

foo <- function(x) {  
  array <- c(df[1,x])
  
  for (i in 2:nrow(df)) {
    if (df[i, 1] == df[i-1, 1 ]) {
      array <- append(array, df[i, x])
    } else {
      array <- append(array, rep(NA, 5))
      array <- append(array, df[i, x])
    }
  }
  # 待修改：重命名array
  print(array)
}

foo("Score1")

解决方法：使用assign()函数动态赋值 #

R中不能直接用字符串作为变量名赋值，需要用assign()函数实现动态命名。该函数可以将值绑定到指定名称的变量上，还能指定变量存放的环境（比如全局环境.GlobalEnv）。

修改后的函数如下：

foo <- function(x) {  
  array <- c(df[1,x])
  
  for (i in 2:nrow(df)) {
    if (df[i, 1] == df[i-1, 1 ]) {
      array <- append(array, df[i, x])
    } else {
      array <- append(array, rep(NA, 5))
      array <- append(array, df[i, x])
    }
  }
  
  # 拼接目标变量名
  var_name <- paste0("array_", x)
  # 将array赋值到全局环境的目标变量
  assign(var_name, array, envir = .GlobalEnv)
  # 删除函数内部的临时array变量
  rm(array)
  # 打印提示信息
  cat("已生成变量：", var_name, "\n")
}

测试运行：

• 执行foo("Score1")后，全局环境会生成array_Score1变量
• 执行foo("Score2")会生成array_Score2，不会和之前的变量冲突

更优方案：返回命名列表（推荐） #

直接在全局环境生成变量容易污染环境，更符合R函数设计习惯的做法是让函数返回命名列表，后续可以灵活处理结果：

foo <- function(x) {  
  array <- c(df[1,x])
  
  for (i in 2:nrow(df)) {
    if (df[i, 1] == df[i-1, 1 ]) {
      array <- append(array, df[i, x])
    } else {
      array <- append(array, rep(NA, 5))
      array <- append(array, df[i, x])
    }
  }
  
  # 返回带有指定名称的列表
  setNames(list(array), paste0("array_", x))
}

# 单个列处理
result_score1 <- foo("Score1")
# 访问结果：result_score1$array_Score1

# 批量处理多个列
all_scores <- lapply(c("Score1", "Score2", "Score3"), foo)
all_scores <- do.call(c, all_scores)
# 访问所有结果：all_scores$array_Score1、all_scores$array_Score2等

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