Подтвердить что ты не робот

Как создать новую переменную в data.frame на основе условия?

Предположим, что у нас есть dataframe

x   y
1   1
2   4 
4   5 

как вы можете добавить новую переменную в dataframe так, что если x меньше или равно 1, он возвращает "хороший", если x находится между 3 и 5, он возвращает "bad" else, возвращает "fair"

x   y  w
1   1  "good"
2   2   "fair"
5   5   "bad"

Применяется метод, показанный ocram., однако этот здесь не работает.

d1 <- c("e", "c", "a")
d2 <- c("e", "a", "b")

w <- ifelse(d1 == "e" & (d2=="e"), 1, ifelse((d1 == "a") & (d2 =="b"), 2, ifelse(d1 == "e"),3,99))

Любые идеи? Благодаря

4b9b3361

Ответ 1

Одна очевидная и простая возможность - использовать "условия if-else". В этом примере

x <- c(1, 2, 4)
y <- c(1, 4, 5)
w <- ifelse(x <= 1, "good", ifelse((x >= 3) & (x <= 5), "bad", "fair"))
data.frame(x, y, w)

** За дополнительным вопросом в редакции ** Это то, что вы ожидаете?

> d1 <- c("e", "c", "a")
> d2 <- c("e", "a", "b")
> 
> w <- ifelse((d1 == "e") & (d2 == "e"), 1, 
+    ifelse((d1=="a") & (d2 == "b"), 2,
+    ifelse((d1 == "e"), 3, 99)))
>     
> data.frame(d1, d2, w)
  d1 d2  w
1  e  e  1
2  c  a 99
3  a  b  2

Если вы не чувствуете себя комфортно с помощью функции ifelse, вы также можете работать с операторами if и else для таких приложений.

Ответ 2

Если у вас очень ограниченное количество уровней, вы можете попробовать преобразовать y в коэффициент и изменить его уровни.

> xy <- data.frame(x = c(1, 2, 4), y = c(1, 4, 5))
> xy$w <- as.factor(xy$y)
> levels(xy$w) <- c("good", "fair", "bad")
> xy
  x y    w
1 1 1 good
2 2 4 fair
3 4 5  bad