如何在R语言中依据color ID列的值拆分数据框的color列？

这问题我平时处理长转宽需求时经常碰到，用tidyr包的pivot_wider()就能轻松搞定，先给你理清楚完整步骤：

1. 补全输入数据框 #

你提供的输入代码没写完，我假设每个id对应color_ID为"a"和"b"各一条记录，完整输入代码如下：

# 构建输入数据框
df.input <- data.frame(
  id = c(1, 1, 2, 2),
  color_ID = c("a", "b", "a", "b"),
  color = c("red", "blue", "green", "yellow"),
  stringsAsFactors = FALSE # 避免默认因子类型带来的问题
)

2. 核心解决方案（推荐用tidyr） #

使用pivot_wider()将长格式数据转为宽格式，同时直接生成你需要的color.1和color.2列：

library(tidyr)

# 转换并生成目标列
df.output <- df.input %>%
  pivot_wider(
    id_cols = id,               # 按id分组
    names_from = color_ID,      # 用color_ID的取值生成新列名
    values_from = color,        # 用color列的内容填充新列
    names_glue = "color.{ifelse(color_ID == 'a', 1, 2)}" # 把a映射为1，b映射为2
  )

运行后得到的df.output就是你要的结果：

id color.1 color.2
1  1     red    blue
2  2   green  yellow

代码解释 #

• id_cols = id：指定按id列进行分组，确保每个id只保留一行
• names_from = color_ID：用color_ID的取值（"a"和"b"）作为新列名的基础
• values_from = color：把color列的内容填充到对应的新列中
• names_glue：自定义新列名的格式，通过ifelse把"a"替换成1，"b"替换成2，生成color.1和color.2

3. 备选方案（用reshape2包） #

如果你习惯用老版本的reshape2包，也可以用dcast()实现：

library(reshape2)

# 先转成宽格式
df.output2 <- dcast(df.input, id ~ color_ID, value.var = "color")
# 手动修改列名
colnames(df.output2) <- c("id", "color.1", "color.2")

这个方法也能得到完全一致的结果，适合还在使用旧版工具的场景。

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