R语言两列值匹配时更新val列为对应分组首行值的实现问题

以下是三种主流实现方案，均可满足按a、b两列组合分组，将分组内所有val值替换为分组首行val的需求：

方法1：使用dplyr包（tidyverse生态，代码可读性最高） #

按指定列分组后调用first()函数取分组内首个val值覆盖原列即可：

# 加载依赖包
library(dplyr)

df_result <- df %>%
  group_by(a, b) %>%
  mutate(val = first(val)) %>%
  ungroup() # 处理完成后取消分组，避免后续操作受分组状态影响

如果需要保留原始val列，可将赋值目标改为新列名，例如mutate(val_first = first(val))。

方法2：Base R原生实现（无需安装额外依赖） #

用ave()函数完成分组计算：

df$val <- ave(df$val, df$a, df$b, FUN = function(x) x[1])

该函数会按传入的分组列对目标向量拆分，应用自定义函数取每组第一个值后，返回和原向量长度完全匹配的结果，直接赋值即可。

方法3：data.table实现（适合十万行以上的大数据集，性能最优） #

library(data.table)
setDT(df) # 转换为data.table格式
df[, val := first(val), by = .(a, b)] # 按分组原地修改val列，无额外拷贝开销

处理完成后的数据集输出如下：

a  b val
1 A1 B1  10
2 A1 B1  10
3 A2 B2  30
4 A2 B2  30

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