Funktion zum Zurückgeben der ersten Zeilen, in denen die Summe der Spaltenwerte größer als eine Zahl ist. R

Question

Ich möchte die ersten Zeilen eines Datenrahmens unter der Bedingung auswählen, dass die Summe der Werte einer Spalte größer als eine bestimmte Zahl ist . Nehmen Sie zum Beispiel die Datenautos. Angenommen, ich möchte Zeilen basierend auf Werten in der zweiten Spalte auswählen. Wenn meine Nummer 7 ist, werden die erste und die zweite Zeile ausgewählt, die Werte in der zweiten Spalte für die ersten zwei Zeilen sind (2 + 10)> 7); Wenn die Anzahl 34 ist, werden die ersten 4 Zeilen ausgewählt werden, (2 + 10 + 4 + 22> 34)

habe ich versucht, die folgende, die das alle Zeilen unabhängig von dem Eingabewert

foo<-function (z) { 
    for (i in 1: nrow (cars)) { 
    if(sum(cars[i,2])<z) { 
    sum(cars[1:i,2]) 
    } else { 
    return (cars[1:i,]) 
    } 
    } 
    return (cars[1:i,]) 
} 

I kehrt versuchte auch unter Verwendung while, aber endete mit nur dem ersten Wert, der endlos zurückgebracht wurde. wäre viel besser

Answer 1

Wir können ein cumsum mit which

cars[seq(which.max(cumsum(cars$dist)>34)),] 

tun, und dies kann in eine Funktion

f1 <- function(dat, col, thresh){ 
     dat[seq(which.max(cumsum(dat[[col]])> thresh)),] 
} 

f1(cars, "dist", 34) 
# speed dist 
#1  4 2 
#2  4 10 
#3  7 4 
#4  7 22 

f1(cars, "dist", 7) 
# speed dist 
#1  4 2 
#2  4 10