R data.table中如何将指定行拆分复制到新行并按条件清空列

实现代码 #

library(data.table)
# 原始数据
dt <- data.table(t0.id=1:2,t0.V1=c("a","e"),t0.V2=c("b","f"),t1.id=3:4,t1.V1=c("c","g"),t1.V2=c("d","h"))

# 批量识别前缀列，无需手动罗列列名，适配多列业务场景
t0_cols <- grep("^t0\\.", names(dt), value = TRUE)
t1_cols <- grep("^t1\\.", names(dt), value = TRUE)

# 全量按条件处理，无行索引操作
res <- rbindlist(list(
  # 拆分后第一行：保留t0前缀列，清空所有t1前缀列
  dt[t0.id == 1, .SD, .SDcols = names(dt)][, (t1_cols) := NA],
  # 拆分后第二行：保留t1前缀列，清空所有t0前缀列
  dt[t0.id == 1, .SD, .SDcols = names(dt)][, (t0_cols) := NA],
  # 无需拆分的行直接保留
  dt[t0.id != 1]
), use.names = TRUE)

# 查看结果
res

输出验证 #

t0.id t0.V1 t0.V2 t1.id t1.V1 t1.V2
1:     1     a     b    NA  <NA>  <NA>
2:    NA  <NA>  <NA>     3     c     d
3:     2     e     f     4     g     h

如果需要和你示例里一样显示为完全空白，可以执行options(datatable.print.na = "")后再打印结果。

方案说明 #

• 全程基于data.table原生的条件筛选和批量列赋值实现，无行索引遍历，针对万级以上大表也能保持很高的处理效率
• 前缀列用正则自动匹配，后续新增t0、t1开头的字段不需要修改代码，适配性强
• 拆分逻辑完全基于t0.id字段判断，你可以自行修改t0.id == 1的条件匹配需要拆分的行
• 不会修改原始数据表，所有操作都在新生成的对象上执行

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