首页 > 解决方案 > 在 R 中使用带有 mutate 的循环来对列名部分匹配的列求和

问题描述

df <- data.frame(x_1_jr=c(1,2,3,4), x_2_jr=c(1,2,3,4), y_1_jr=c(4,3,2,1), y_2_jr=c(4,3,2,1)

  x_1_jr x_2_jr y_1_jr y_2_jr
1      1      1      4      4
2      2      2      3      3
3      3      3      2      2
4      4      4      1      1

我正在尝试生成具有相同列名后缀的 x 和 y 之和的新变量,即

df <- df %>% mutate(z_1_jr= x_1_jr + y_1_jr)

  x_1_jr x_2_jr y_1_jr y_2_jr z_1_jr
1      1      1      4      4      5
2      2      2      3      3      5
3      3      3      2      2      5
4      4      4      1      1      5

我可以为每个变量组合写出这个,但是我有大量变量(每个 x 和 y 组>50),并且想使用循环......但是,我对 R 比较陌生,我不知道从哪里开始!

有人可以帮忙吗?谢谢!

编辑:为了更加清楚,数据集包含其他非数字变量。有 >700 列(来自大型调查)。例如,x_1_jr 表示 1 岁的男性个体的数量,y_1_jr 1 岁的女性个体的数量。我试图为每个年龄组计算一个总数(男性加女性 1 年)。

一个

标签: rloopsdplyrnames

解决方案


一个dplyr选项purrr可能是:

df %>%
 bind_cols(map_dfc(.x = unique(sub(".*?_", "_", names(df))),
                   ~ df %>%
                    transmute(!!paste0("z", .x) := rowSums(select(., ends_with(.x))))))

  x_1_jr x_2_jr y_1_jr y_2_jr z_1_jr z_2_jr
1      1      1      4      4      5      5
2      2      2      3      3      5      5
3      3      3      2      2      5      5
4      4      4      1      1      5      5

推荐阅读