R语言按指定列名拆分数据框失败：生成NULL列表求助

我有一个含100列的大型数据集，列名分为A、B、C、D四类，想按列名将其拆分为四个独立的数据框。作为编程新手，折腾了好几个小时，尝试用for循环创建名为split_dfs的列表，预期里面是对应分组的数据框，但print(split_dfs)结果为NULL，元素没成功加进去。

示例数据 #

df <- data.frame(
  A = c(0.6, 0.9, 0.18),
  B = c(0.1, 0.81, 0.7),
  C = c(0.3, 0.5, 0.79),
  B = c(0.4, 0.73, 0.3),
  D = c(0.2, 0.3, 0.5),
  A = c(0.1, 0.8, 0.4),
  C = c(0.3, 0.5, 0.7),
  C = c(0.9, 0.8, 0.4),
  B = c(0.9, 0.1, 0.4)
)

预期结果 #

# group A
#      A
# 1 0.60
# 2 0.90
# 3 0.18
# 4 0.10
# 5 0.80
# 6 0.40

# group B
#      B
# 1 0.10
# 2 0.81
# 3 0.70
# 4 0.40
# 5 0.73
# 6 0.30
# 7 0.90
# 8 0.10
# 9 0.40

# group C
#      C
# 1 0.30
# 2 0.50
# 3 0.79
# 4 0.30
# 5 0.50
# 6 0.70
# 7 0.90
# 8 0.80
# 9 0.40

# group D
#      D
# 1 0.20
# 2 0.30
# 3 0.50

所用错误代码 #

col_groups = list(
  c("A"),
  c("B"),
  c("C"),
  c("D")
)

split_dfs = list()
for(i in names(col_groups)){
  selected_cols = col_groups[[i]]
  split_dfs[[i]] = new_df[, selected_cols, drop = FALSE]
}

修正方案 #

1. 修复你的for循环代码 #

你的代码核心问题是：col_groups是无命名的列表，names(col_groups)返回NULL，导致循环根本没执行，所以split_dfs一直是空列表。

修正后的代码：

# 给分组列表命名，方便后续索引
col_groups <- list(A = "A", B = "B", C = "C", D = "D")
split_dfs <- list()

# 直接遍历分组名称
for(group_name in names(col_groups)){
  # 筛选所有列名等于当前分组的列
  selected_cols <- names(df) == group_name
  # 提取列并转为数据框（drop=FALSE确保结果是数据框而非向量）
  split_dfs[[group_name]] <- df[, selected_cols, drop = FALSE]
}

# 查看结果
print(split_dfs)

2. 更简洁的R内置函数方法（推荐） #

R有专门的split.default()函数，可以直接按列名分组拆分数据框，一行代码搞定，比循环更高效简洁：

# 直接按列名分组，返回命名列表，每个元素是对应分组的数据框
split_dfs <- split.default(df, names(df))

# 查看结果
print(split_dfs)

如果需要把每个分组的数据框单独提取出来（比如单独存为df_A、df_B），可以用list2env()：

list2env(split_dfs, envir = .GlobalEnv)
# 之后直接调用df_A、df_B等即可

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