r中在一定條件下的組累計和

Question

在具有相同的m1和n1的組內，如果x1不為0，則需要計算x1累積和，如果y1不為0，則需要計算y1累積和。可以提供任何幫助。

 d <- data.frame () 
    for ( m1 in 2: 3) { 
         for (n1 in 2: 3){ 
              for (x1 in 0: m1) { 
                   for (y1 in 0: n1) {                                                                                                         

                  d<-rbind(d, c(m1, n1, x1,y1)) 
    }}}} 

因此對於m1=2 ， n1=2 ， x1的累積和應為9， y1為9，對於m1=2 ， n1=3 ， x1的累積和應為12， y1為18。

Answer 1

這是你想要的？ （使用data.table ）。 而且我認為您想對所有元素求和，而不是對所有元素進行累加和返回。

# after running your code, I had to rename it
names(d) <- c("m1", "n1", "x1", "y1")
require(data.table)
dt <- data.table(d)
setkey(dt, m1, n1)
out <- dt[, list(s.x1=sum(x1), x.y1=sum(y1)),by="m1,n1"]
> out
#    m1 n1 s.x1 x.y1
# 1:  2  2    9    9
# 2:  2  3   12   18
# 3:  3  2   18   12
# 4:  3  3   24   24