R For循环使用if语句应用格式

Question

假设我有一个看起来像这样的for循环。

person <- c("Mike", "Kim", "Fred", "Steve", "Quail", "bean")

sleep <- c(5, 10, 3, 7, 1, 5)


dat <- data.frame(person, sleep)

for (i in 1:length(dat$person)) {
 if (dat$sleep[i] >= 7) {
   dat$fatigue[i] <- 0
 }
 if (dat$sleep[i] > 5 & dat$sleep[i] < 7) {
   dat$fatigue[i] <- 1
 }
 if (dat$sleep[i] >= 3 & dat$sleep[i] <= 5) {
   dat$fatigue[i] <- 2
 }
 if (dat$sleep[i] < 3) {
  dat$fatigue[i] <- 3
 }
}

我如何使用R的apply函数用if语句写这个for循环？

我很擅长使用这个，任何帮助将不胜感激。

请让我知道，如果你有任何问题。

谢谢。

Answer 1

我们可以尝试使用case_when库中的dplyr

dat$fatigue <- case_when(
    dat$sleep < 3 ~ 3,
    dat$sleep >= 3 & dat$sleep <= 5 ~ 2,
    dat$sleep > 5 & dat$sleep < 7 ~ 1,
    dat$sleep >= 7 ~ 0,
    TRUE ~ -1
)

请注意，case_when()与R中的许多函数一样，已经过矢量化。因此，您不需要使用显式的for循环。

Answer 2

如果你想使用嵌套的ifelse，这个解决方案可以使用baseR

person <- c("Mike", "Kim", "Fred", "Steve", "Quail", "bean")

sleep <- c(5, 10, 3, 7, 1, 5)

dat <- data.frame(person, sleep)

dat$fatigue <- ifelse(dat$sleep < 3,3,ifelse(dat$sleep >= 3 & dat$sleep <= 5, 2,ifelse(dat$sleep > 5 & dat$sleep < 7,1,ifelse(dat$sleep >= 7, 0,"NA"))))
)