创建多个总和

时间:2018-09-18 16:46:23

标签: r dplyr rowsum

Ciao, 这是一个可复制的示例。

   df <- data.frame("STUDENT"=c(1,2,3,4,5),
                     "TEST1A"=c(NA,5,5,6,7),
                     "TEST2A"=c(NA,8,4,6,9),
                     "TEST3A"=c(NA,10,5,4,6),
                     "TEST1B"=c(5,6,7,4,1),
                     "TEST2B"=c(10,10,9,3,1),
                     "TEST3B"=c(0,5,6,9,NA),
                     "TEST1TOTAL"=c(NA,23,14,16,22),
                     "TEST2TOTAL"=c(10,16,15,12,NA))

我具有从STESTENT到TEST3B的列,并想要创建TEST1TOTAL TEST2TOTAL。 TEST1TOTAL = TEST1A + TEST2A + TEST3A,依此类推。如果TEST1A TEST2A TEST3A中缺少任何分数,则TEST1TOTAL为NA。

这是我的尝试,但是有没有用更少的代码行的解决方案?因为在这里,我需要多次编写此行,因为测试A到O都是如此。

TEST1TOTAL=rowSums(df[,c('TEST1A', 'TEST2A', 'TEST3A')], na.rm=TRUE)

3 个答案:

答案 0 :(得分:2)

尝试:

library(dplyr)
df %>%
        mutate(TEST1TOTAL = TEST1A+TEST2A+TEST3A,
               TEST2TOTAL = TEST1B+TEST2B+TEST3B)

df %>%
        mutate(TEST1TOTAL = rowSums(select(df, ends_with("A"))),
               TEST2TOTAL = rowSums(select(df, ends_with("B"))))

答案 1 :(得分:2)

仅使用R基本函数:

output <- data.frame(df1, do.call(cbind, lapply(c("A$", "B$"), function(x) rowSums(df1[, grep(x, names(df1))]))))

自定义姓氏:

> colnames(output)[(ncol(output)-1):ncol(output)] <- c("TEST1TOTAL", "TEST2TOTAL")
> output
  STUDENT TEST1A TEST2A TEST3A TEST1B TEST2B TEST3B TEST1TOTAL TEST2TOTAL
1       1     NA     NA     NA      5     10      0         NA         15
2       2      5      8     10      6     10      5         23         21
3       3      5      4      5      7      9      6         14         22
4       4      6      6      4      4      3      9         16         16
5       5      7      9      6      1      1     NA         22         NA

答案 2 :(得分:1)

我认为,Jilber Urbina's solution是您的理想之选。出于完整性考虑(并且因为我学到了一些知识),这是一种整齐的方法,可以针对任意数量的测试按测试编号获得总分。

优点是您无需指定测试的标识符(除了已编号或带有结尾字母的标识符),并且相同的代码可用于任意数量的测试。

library(tidyverse)

df_totals <- df %>%
    gather(test, score, -STUDENT) %>%                    # Convert from wide to long format
    mutate(test_num = paste0('TEST', ('[^0-9]', '', test),
                             'TOTAL'),                   # Extract test_number from variable
           test_let = gsub('TEST[0-9]*', '', test)) %>%  # Extract test_letter (optional)
    group_by(STUDENT, test_num) %>%                      # group by student + test
    summarize(score_tot = sum(score)) %>%                # Sum score by student/test
    spread(test_num, score_tot)                          # Spread back to wide format

df_totals

# A tibble: 5 x 4
# Groups:   STUDENT [5]
  STUDENT TEST1TOTAL TEST2TOTAL TEST3TOTAL
    <dbl>      <dbl>      <dbl>      <dbl>
1       1         NA         NA         NA
2       2         11         18         15
3       3         12         13         11
4       4         10          9         13
5       5          8         10         NA

如果您也想获得个人分数,只需将总计与原始分数相加即可:

left_join(df, df_totals, by = 'STUDENT')
  STUDENT TEST1A TEST2A TEST3A TEST1B TEST2B TEST3B TEST1TOTAL TEST2TOTAL TEST3TOTAL
1       1     NA     NA     NA      5     10      0         NA         NA         NA
2       2      5      8     10      6     10      5         11         18         15
3       3      5      4      5      7      9      6         12         13         11
4       4      6      6      4      4      3      9         10          9         13
5       5      7      9      6      1      1     NA          8         10         NA