R语言中将单列dataframe转为向量且保留列名作为向量名的方法

基础实现（直接生成同列名向量） #

用R内置的assign()函数即可实现自动匹配列名生成对应向量，无需手动指定变量名：

# 示例单列dataframe
df <- data.frame(variable_name = c(2,4,6,8,10))

# 直接生成与列名同名的向量
assign(colnames(df)[1], df[[1]])

# 此时即可直接调用与列名同名的向量
variable_name

备选快速实现 #

也可以通过list2env()实现，代码更简洁：

df <- data.frame(variable_name = c(2,4,6,8,10))
list2env(as.list(df), envir = .GlobalEnv)

函数场景适配 #

如果是在自定义函数中使用，需要指定赋值的目标环境，避免变量仅生成在函数内部环境中无法调用，示例函数如下：

# 封装好的转换函数
single_col_to_vec <- function(input_df, target_env = parent.frame()) {
  # 校验输入为单列dataframe
  if (ncol(input_df) != 1) {
    stop("输入必须为仅包含1列的dataframe")
  }
  # 提取原列名
  vec_name <- colnames(input_df)[1]
  # 赋值到指定环境，默认是调用函数的上层环境
  assign(vec_name, input_df[[1]], envir = target_env)
  # 无感知返回向量值，不强制打印
  invisible(input_df[[1]])
}

函数调用示例 #

# 构造测试dataframe
test_df <- data.frame(sales = c(230, 510, 170, 490))

# 调用转换函数
single_col_to_vec(test_df)

# 此时已自动生成名为sales的向量
sales

注意：如果需要将生成的向量存入指定环境，只需修改target_env参数传入对应环境对象即可。

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