如何对数据框中标题含指定字符串的列行求和并添加新列?
按列名匹配实现DataFrame行求和并添加新列
问题场景
有一个DataFrame,部分列名包含字符串"hsehold",另一部分包含"away"(例如:hsehold1、hsehold2、hsehold3、away1、away2、away3),需要为该DataFrame添加新列,值为所有标题含"hsehold"的列的行求和结果。
错误尝试及提示
尝试的代码:
df$newCol.hsehold <- rowSums(df, na.rm = TRUE, select(matches("hsehold")))
df$newCol.hsehold <- rowSums(df, na.rm = TRUE, select(contains("hsehold")))
收到错误提示:contains() must be used within a selecting function.(翻译:contains()必须在选择函数中使用)
正确解法
方法1:dplyr语法实现
使用mutate结合across指定目标列,符合tidyverse风格:
library(dplyr) # 用contains匹配列名 df <- df %>% mutate(newCol.hsehold = rowSums(across(contains("hsehold")), na.rm = TRUE)) # 或用matches正则匹配(支持更灵活的列名规则) df <- df %>% mutate(newCol.hsehold = rowSums(across(matches("hsehold")), na.rm = TRUE))
方法2:基础R实现
无需额外包,直接通过列名筛选实现:
# 筛选列名包含"hsehold"的列索引 hsehold_cols <- grepl("hsehold", colnames(df)) # 行求和并添加新列 df$newCol.hsehold <- rowSums(df[, hsehold_cols], na.rm = TRUE)
错误原因
此前的写法错误在于:rowSums不支持直接传入select这类dplyr选择函数作为参数。select需要在dplyr的管道操作或数据框选择上下文中使用,而rowSums仅接受数据框/矩阵作为核心输入,额外参数仅包含na.rm这类配置项。
内容的提问来源于stack exchange,提问作者Sarah
相关产品推荐
相关产品推荐

