问题描述
df <- data.frame('a'=c(1,2,3,4,5),'M1'=c(1,20,50))
现在我想创建一个额外的 179 列 (M2 - M180),它取决于之前 Mx 列的总和。例如,在这种情况下,M2 应该是所有先前 Mx 值的总和,即 M1。而M3应该是M1+M2之和,以此类推,一直到M180。
前三个 Mx 列的预期结果应该是:
Expected_res <- data.frame('a'=c(1,50),'M2'=c(1,'M3'=c(2,40,6,8,100))
基本上我想要相当于 Sum($E1:E1) 的 excel,然后将其拖出 180 列,当然是每一行。
感谢任何帮助。
解决方法
您可以使用 for
循环:
for(i in 2:180) {
df[paste0('M',i)] <- rowSums(df[-1])
}
-1
忽略第一列 (a
) 并计算所有其他列的总和以创建新列。
这是前 5 列的输出。
df
# a M1 M2 M3 M4 M5
#1 1 1 1 2 4 8
#2 2 20 20 40 80 160
#3 3 3 3 6 12 24
#4 4 4 4 8 16 32
#5 5 50 50 100 200 400
,
使用tidyverse
library(tidyverse)
df <- data.frame('a'=c(1,2,3,4,5),'M1'=c(1,20,50))
n <- 5
bind_cols(df,map_dfc(seq_len(n),~transmute(df,!!paste0("M",.x + 1) := M1 * 2 ^ (.x - 1))))
#> a M1 M2 M3 M4 M5 M6
#> 1 1 1 1 2 4 8 16
#> 2 2 20 20 40 80 160 320
#> 3 3 3 3 6 12 24 48
#> 4 4 4 4 8 16 32 64
#> 5 5 50 50 100 200 400 800
由 reprex package (v1.0.0) 于 2021 年 2 月 2 日创建
使用data.table
library(data.table)
df <- data.frame('a'=c(1,50))
setDT(df)
n <- 5
col_names <- paste0("M",seq_len(n) + 1)
df[,(col_names) := lapply(seq_len(n),function(x) M1 * 2 ^ (x - 1))]
df
#> a M1 M2 M3 M4 M5 M6
#> 1: 1 1 1 2 4 8 16
#> 2: 2 20 20 40 80 160 320
#> 3: 3 3 3 6 12 24 48
#> 4: 4 4 4 8 16 32 64
#> 5: 5 50 50 100 200 400 800
由 reprex package (v1.0.0) 于 2021 年 2 月 2 日创建