Wie Wert einer Spalte auf der Basis eines Spaltenindex

Question

Mit einem Datenrahmen zuweisen Ich möchte einen berechneten Wert auf einer gegebenen Spaltenindex

df <- data.frame(a = c(2,4,7,3,5,3), b = c(8,3,8,2,6,1)) 

> df 
    a b 
1 2 8  
2 4 3 
3 7 8 
4 3 2 
5 5 6 
6 3 1 

max <- apply(df, 1, which.max) 
> max 
[1] 2 1 2 1 2 1 

addition <- apply(df, 1, sum) 
> addition 
[1] 10 7 15 5 11 4 

Dann einige Operation basierend zuweisen, die ich nicht heraus kann mit dem folgenden Ergebnis zugewiesen werden df2

> df2 
    a b 
1 2 10 
2 7 3 
3 7 15 
4 5 2 
5 5 11 
6 4 1 

hoch schätzen Sie Ihre Ideen und Ihre Hilfe. Danke

Answer 1

Sie können auch ifelse direkt vektorisiert verwenden:

with(df, cbind.data.frame(a = ifelse(a > b, a + b, a), b = ifelse(a > b, b, a + b))); 
# a b 
#1 2 10 
#2 7 3 
#3 7 15 
#4 5 2 
#5 5 11 
#6 4 1 

Answer 2

Sie cbind verwenden können Sie Ihre ausgewählten Spalten für jede Zeile zuzugreifen:

df2 = df 
df2[cbind(1:nrow(df2),max)] = addition 
df2 
    a b 
1 2 10 
2 7 3 
3 7 15 
4 5 2 
5 5 11 
6 4 1 

Hier cbind liefert eine Matrix von 2 Spalten und 6 Zeilen, die wir verwenden, um den Datenrahmen unter Verwendung von Matrix subsetting der Teilmenge.