berechnen Mittelwert akkumulierte mit Bedingung

Question

Ich habe einen data.table wie diese

a <- data.table(
    A = c(1:6), 
    B = c(1,2,2,3,4,2), 
    C = c("t","t","f","t","f","t") 
) 
    A B C 
1: 1 1 t 
2: 2 2 t 
3: 3 2 f 
4: 4 3 t 
5: 5 4 f 
6: 6 2 t 

In jeder Reihe, ich möchte den Mittelwert aller vorherigen Zeilen berechnen, aber mit der Bedingung, dass sie die Zeile mit C == f ausschließen . Das Ergebnis sollte

A B C D E 
1: 1 1 t NA NA 
2: 2 2 t 1 1 
3: 3 2 f 1.5 1.5 
4: 4 3 t 1.5 1.5 
5: 5 4 f 3.5 3 
6: 6 2 t 3.5 3 

sein Wie kann ich das tun?

Answer 1

Meinst du den kumulativen Mittelwert? Dann sollte dies tun, was Sie verlangen:

a[C != "f", lapply(.SD, FUN = function(x) {cumsum(x)/seq_along(x)}), 
    .SDcols = c("A", "B")] 

      A B 
1: 1.000000 1.0 
2: 1.500000 1.5 
3: 2.333333 2.0 
4: 3.250000 2.0