Entfernen Sie letzte n Zeilen für 0 durch ID und dann Schwanz letzte Elemente durch ID in R

Question

final dataset Ich habe einen Datensatz als Anhang.

Ich möchte alle Last 0s von Id in
binär zu entfernen, und wählen Sie dann Schwanz 3 für von id. (Anmerkung: für Id 2 nur Restng 2
w5 und w6). Es wird wie der endgültige Datensatz sein.

Gibt es einen effizienten Weg, dies zu tun?

Answer 1

Wie wäre es mit so etwas?

ID = c(rep(1, 8), rep(2, 4), rep(3, 8)) 
week = c(2:9, 5:8, 13:20) 
binary = c(0, 1, 1, 1, 1, 0, 0, 0, 1, 1, 0, 0, 1, 1, 1, 0, 0, 1, 0, 0) 

dataset = data.frame(ID, week, binary) 

> dataset 
    ID week binary 
1 1 2  0 
2 1 3  1 
3 1 4  1 
4 1 5  1 
5 1 6  1 
6 1 7  0 
7 1 8  0 
8 1 9  0 
9 2 5  1 
10 2 6  1 
11 2 7  0 
12 2 8  0 
13 3 13  1 
14 3 14  1 
15 3 15  1 
16 3 16  0 
17 3 17  0 
18 3 18  1 
19 3 19  0 
20 3 20  0 

Lösung:

# Split dataset by ID 
byGroupDf = split(dataset, ID) 

# For each grouped dataset, take only rows up till the last row where ID not equal to 0 
byGroupFinal = lapply(byGroupDf, function(x) tail(x[1:max(which(x$binary != 0)),], 3)) 

# Combine the grouped datasets 
FinalDf = do.call(rbind, byGroupFinal) 

> FinalDf 
    ID week binary 
1.3 1 4  1 
1.4 1 5  1 
1.5 1 6  1 
2.9 2 5  1 
2.10 2 6  1 
3.16 3 16  0 
3.17 3 17  0 
3.18 3 18  1