Dataframe utworzyć nową kolumnę na podstawie innych kolumn

Question

Mam dataframe:

df <- data.frame('a'=c(1,2,3,4,5), 'b'=c(1,20,3,4,50)) 
df 
    a b 
1 1 1 
2 2 20 
3 3 3 
4 4 4 
5 5 50 

i chcę utworzyć nową kolumnę na podstawie istniejących kolumn. Coś takiego:

if (df[['a']] == df[['b']]) { 
    df[['c']] <- df[['a']] + df[['b']] 
} else { 
    df[['c']] <- df[['b']] - df[['a']] 
} 

Problem polega na tym, że warunek if sprawdzana jest tylko dla pierwszego rzędu ... Jeśli utworzyć funkcję z powyższym stwierdzeniem if następnie używam apply() (lub mapply() ...) to jest to samo.

W Pythonie/pandy mogę użyć tego:

df['c'] = df[['a', 'b']].apply(lambda x: x['a'] + x['b'] if (x['a'] == x['b']) \ 
    else x['b'] - x['a'], axis=1) 

Chcę coś podobnego w R. Tak więc wynik powinien wyglądać następująco:

a b c 
1 1 1 2 
2 2 20 18 
3 3 3 6 
4 4 4 8 
5 5 50 45 

Answer 1

Jedną z opcji jest ifelse który jest wersja vectorized z if/else. Jeśli robimy to dla każdego wiersza, to if/else, jak pokazano w postach Panda OP można zrobić w pętli for lub lapply/sapply, ale byłoby to nieefektywne w R.

df <- transform(df, c= ifelse(a==b, a+b, b-a)) 
df 
# a b c 
#1 1 1 2 
#2 2 20 18 
#3 3 3 6 
#4 4 4 8 
#5 5 50 45 

---

Można to inaczej zapisać jako

df$c <- with(df, ifelse(a==b, a+b, b-a)) 

stworzyć 'c' kolumny w oryginalnym zbiorze

---

Jak PO chce podobną opcję w R używanie if/else

df$c <- apply(df, 1, FUN = function(x) if(x[1]==x[2]) x[1]+x[2] else x[2]-x[1]) 

Answer 2

Tu jest nieco bardziej skomplikowane metody algebraiczne:

df$c <- with(df, b + ((-1)^((a==b)+1) * a)) 

df 
    a b c 
1 1 1 2 
2 2 20 18 
3 3 3 6 
4 4 4 8 
5 5 50 45 

Chodzi o to, że operator „minus” jest włączone lub wyłączone na podstawie testu a==b.

Answer 3

Rozwiązanie z apply

myFunction <- function(x){ 
    a <- x[1] 
    b <- x[2] 
    #further values ignored (if there are more than 2 columns) 
    value <- if(a==b) a + b else b - a 
    #or more complicated stuff 
    return(value) 
} 

df$c <- apply(df, 1, myFunction) 

Answer 4

Jeśli chcesz metodę stosuje się, po czym inny sposób z mapply byłoby utworzyć funkcję i zastosowanie go,

fun1 <- function(x, y) if (x == y) {x + y} else {y-x} 
df$c <- mapply(fun1, df$a, df$b) 
df 
# a b c 
#1 1 1 2 
#2 2 20 18 
#3 3 3 6 
#4 4 4 8 
#5 5 50 45 

Answer 5

Korzystanie dplyr pakiet:

library(dplyr) 

df <- df %>% 
    mutate(c = if_else(a == b, a + b, b - a)) 

df 
# a b c 
# 1 1 1 2 
# 2 2 20 18 
# 3 3 3 6 
# 4 4 4 8 
# 5 5 50 45