在 R 中使用带有条件的 ifelse 进行变异

Question

我想为温度创建数据框

时间范围 : 2020-01-01 ~ 2020-12-31
温度 ：
2020-01-01 ~ 2020-02-29 和 2020-10-01 ~ 2020-12-31 : 10 ~ 24 之间的随机整数
2020-03-01 ~ 2020-05-31 : 5 ~ 17 之间的随机整数
2020-06-01 ~ 2020-09-31 : 1 ~ 8 之间的随机整数

我的代码是

library(tidyverse)
library(lubridate)
library(summarytools)
library(dplyr)

start <- as.POSIXct("2020-01-01", "%Y-%m-%d", tz = "UTC")
end   <- as.POSIXct("2020-12-31", "%Y-%m-%d", tz = "UTC")

remT <- seq(start, end, by = "1 day") 
date_df <- as.data.frame(remT)
date_df <- setNames(date_df, c("Date")) 
date_df <- date_df %>% arrange(Date)

cond_1 <- for(i in date_df$Date){(i >= as.Date('2020-01-01') && i <= as.Date('2020-02-29')) || (i >= as.Date('2020-10-01') && i <= as.Date('2020-12-31'))}
cond_2 <- for(j in date_df$Date)(j >= as.Date('2020-06-01') && j <= as.Date('2020-09-30'))

x <- sample(10:24, 152, replace=TRUE)
y <- sample(1:8, 122, replace=TRUE)
z <- sample(5:17, 92, replace=TRUE)


date_df <- date_df %>%
  mutate(Test = 
           ifelse(cond_1, x,
                     ifelse(cond_2, y , z)
                  )
         )

但是，不幸的是，此代码返回了错误消息

错误： mutate()列Test 。 ℹ Test = ifelse(cond_1, x, ifelse(cond_2, y, z)) 。 ℹ Test必须是大小 366 或 1，而不是 0。

如何解决这个错误？

Answer 1

mutate需要 1 个或一组完整的值。 case_when或ifelse或if_else将根据给定条件处理在哪里替换值。 所以对所有条件采样 366 个值。

最好的方法是使用n()代替任何 value 。 此外，您可以在一根管道中完成此操作

library(tidyverse)
library(lubridate, warn.conflicts = F)

set.seed(123)
seq.Date(as.Date('2020-01-01'), as.Date('2020-12-31'), by = 1) %>%
  as.data.frame() %>% setNames('Date') %>%
  mutate(temp = case_when(month(Date) %in% c(1:2, 10:12)~ sample(10:24, n(), T),
                          month(Date) %in% c(3:5) ~ sample(5:17, n(), T),
                          TRUE ~ sample(1:8, n(), T))) -> df

head(df, 5)
#>         Date temp
#> 1 2020-01-01   24
#> 2 2020-01-02   24
#> 3 2020-01-03   12
#> 4 2020-01-04   23
#> 5 2020-01-05   12

tail(df, 5)
#>           Date temp
#> 362 2020-12-27   15
#> 363 2020-12-28   19
#> 364 2020-12-29   19
#> 365 2020-12-30   15
#> 366 2020-12-31   21

^{由reprex 包( v2.0.0 ) 于 2021 年 6 月 17 日创建}

Answer 2

您可以将sapply与switch和sample ：

library(lubridate)
start <- as.POSIXct("2020-01-01", "%Y-%m-%d", tz = "UTC")
end   <- as.POSIXct("2020-12-31", "%Y-%m-%d", tz = "UTC")

remT <- seq(start, end, by = "1 day") 

set.seed(1)
temp <- sapply(as.character(month(remT)), function(m)
  switch(m,
         `1` = , `2` = , `10` = , `11` = , `12` = sample(10:24, 1),
         `3` = , `4` = , `5` = sample(5:17, 1),
         `6` = , `7` = , `8` = , `9` = sample(1:8, 1)))
  

sort(unique(temp[month(remT) %in% c(1:2, 10:12)]))
#R>  [1] 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
sort(unique(temp[month(remT) %in% 3:5]))
#R>  [1]  5  6  7  8  9 10 11 12 13 14 15 16 17
sort(unique(temp[month(remT) %in% 6:9]))
#R> [1] 1 2 3 4 5 6 7 8

稍微快一点的解决方案可能是使用vapply 、 switch和sample.int ：

set.seed(1)
temp <- vapply(as.character(month(remT)), function(m)
  switch(m,
         `1` = , `2` = , `10` = , `11` = , `12` = sample.int(15, 1) + 9L,
         `3` = , `4` = , `5` = sample.int(13, 1) + 4L,
         `6` = , `7` = , `8` = , `9` = sample.int(8, 1)), integer(1))
  

sort(unique(temp[month(remT) %in% c(1:2, 10:12)]))
#R>  [1] 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
sort(unique(temp[month(remT) %in% 3:5]))
#R>  [1]  5  6  7  8  9 10 11 12 13 14 15 16 17
sort(unique(temp[month(remT) %in% 6:9]))
#R> [1] 1 2 3 4 5 6 7 8

否则， ifelse需要与 AnilGoyal 指出的第一个参数相同长度的值。 因此，以下确实有效：

set.seed(1)
temp <- ifelse(
  month(remT) %in% c(1:2, 10:12), sample(10:24, length(remT), TRUE),
  ifelse(month(remT) %in% 3:5, sample(5:17, length(remT), TRUE),
         sample(1:8, length(remT), TRUE)))

sort(unique(temp[month(remT) %in% c(1:2, 10:12)]))
#R>  [1] 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
sort(unique(temp[month(remT) %in% 3:5]))
#R>  [1]  5  6  7  8  9 10 11 12 13 14 15 16 17
sort(unique(temp[month(remT) %in% 6:9]))
#R> [1] 1 2 3 4 5 6 7 8