如何在R的data.table中按行对多列求和生成新列?
在data.table中实现行求和的正确方法
你遇到的问题核心是sum()函数的特性:它会把输入的所有元素合并成一个向量计算全局总和,而非按行单独求和。以下是几种正确实现行求和的方法:
方法1:使用rowSums()(推荐,适合多列场景)
利用rowSums()配合data.table的.SD和.SDcols参数,指定需要计算的列并逐行求和:
library(data.table) adherence[, total := rowSums(.SD), .SDcols = c("ace", "bb", "statin")]
.SD代表当前数据集的子集(无分组时即为整个数据集).SDcols明确指定参与求和的列,避免误包含其他无关列
方法2:直接列相加(适合列数较少的场景)
如果只有少量列需要求和,直接对每行的目标列做加法运算更简洁:
adherence[, total := ace + bb + statin]
验证示例
假设你的adherence数据如下:
adherence <- data.table( ace = c(1, 0, 1, 0, 1), bb = c(1, 1, 0, 0, 1), statin = c(0, 0, 1, 1, 0) )
运行上述任意一种方法后,total列都会得到预期的c(2, 1, 2, 1, 2)。
内容的提问来源于stack exchange,提问作者amatof
相关产品推荐
相关产品推荐

