r - 我想添加数据框的递增连续列

Question

我有一个包含 20 列的数据框。我想先计算前两列的总和，然后计算接下来的三列并继续添加，直到我得到所有 20 列的总和。 DataFrame 只有 0 和 1。如果 k 列的总和超过五，我想将剩余列的值更改为 0。我无法创建循环来执行相同的操作

0 0 0 1 1 0 1 1 1 0 0 0 0 1 0 0
0 1 1 0 0 0 0 0 1 1 0 0 0 0 1 0
1 0 1 0 0 0 1 1 1 0 0 0 0 1 0 1
0 1 1 0 0 0 0 0 0 0 0 0 0 1 1 0
0 1 0 1 0 1 0 0 0 0 0 0 0 1 1 0

例如，在第一行中，因为我们在第 9 列中获得了 sum=5，我想将剩余的值更改为 0，即倒数第三个值更改为 0。

Answer 1

编辑:Base R 解决方案

df <- as.data.frame(dt)


tdf <- data.frame(t(df))

tdf$X1[cumsum(tdf$X1)>5] <- 0
tdf$X2[cumsum(tdf$X2)>5] <- 0
tdf$X3[cumsum(tdf$X3)>5] <- 0
tdf$X4[cumsum(tdf$X4)>5] <- 0
tdf$X5[cumsum(tdf$X5)>5] <- 0


t(tdf)
#>    V1 V2 V3 V4 V5 V6 V7 V8 V9 V10 V11 V12 V13 V14 V15 V16
#> X1  0  0  0  1  1  0  1  1  1   0   0   0   0   0   0   0
#> X2  0  1  1  0  0  0  0  0  1   1   0   0   0   0   1   0
#> X3  1  0  1  0  0  0  1  1  1   0   0   0   0   0   0   0
#> X4  0  1  1  0  0  0  0  0  0   0   0   0   0   1   1   0
#> X5  0  1  0  1  0  1  0  0  0   0   0   0   0   1   1   0

你可以先t()你的dataframe。

并使用cumsum函数和t返回结果

library(data.table)

dt <- fread('0 0 0 1 1 0 1 1 1 0 0 0 0 1 0 0
0 1 1 0 0 0 0 0 1 1 0 0 0 0 1 0
1 0 1 0 0 0 1 1 1 0 0 0 0 1 0 1
0 1 1 0 0 0 0 0 0 0 0 0 0 1 1 0
0 1 0 1 0 1 0 0 0 0 0 0 0 1 1 0')

tdt <- data.table(t(dt))

tdt[,names(tdt):=lapply(.SD,function(x) {x[cumsum(x)>5] <- 0
                                         x})]

t(tdt)
#>    [,1] [,2] [,3] [,4] [,5] [,6] [,7] [,8] [,9] [,10] [,11] [,12] [,13] [,14]
#> V1    0    0    0    1    1    0    1    1    1     0     0     0     0     0
#> V2    0    1    1    0    0    0    0    0    1     1     0     0     0     0
#> V3    1    0    1    0    0    0    1    1    1     0     0     0     0     0
#> V4    0    1    1    0    0    0    0    0    0     0     0     0     0     1
#> V5    0    1    0    1    0    1    0    0    0     0     0     0     0     1
#>    [,15] [,16]
#> V1     0     0
#> V2     1     0
#> V3     0     0
#> V4     1     0
#> V5     1     0

^{由 reprex package 创建于 2020-04-23 (v0.3.0)}