Floor_date 不适用于 mutate 和 ifelse

Question

我试图编写一个概括的聚合函数，其中用户指定聚合级别，或者可以在所有研究日期汇总数据。 floor_date仅转换第一个日期。为什么？我该如何解决？

library(dplyr)
library(lubridate)

sTerm <- "year" # month, bimonth, quarter, season, halfyear and year, custom
sCustom <- "2023-2025"

dfDatasetOutput  <- data.frame(
  valDate=seq(as.Date("2023-01-01"), as.Date("2025-12-01"), by = "month"), 
  cat1=rnorm(36, 3500, 1000),
  cat2=rnorm(36, 2.5, 5)
)

dfDatasetOutput %>%
  mutate(valDate=ifelse(toupper(sTerm)=="CUSTOM", 
                          sCustom, 
                          as.character(floor_date(valDate, sTerm)))) 

# this works just fine
dfDatasetOutput %>%
  mutate(valDate=as.character(floor_date(valDate, sTerm)))

Answer 1

该问题并非源于 floor_date，而是源于您使用 ifelse。根据其手册：

ifelse(test, yes, no)
ifelse returns a value with the same shape as test which is filled
with elements selected from either yes or no depending on whether 
the element of test is TRUE or FALSE.

您的测试是 toupper(sTerm)=="CUSTOM" 这是单个逻辑元素 TRUE 或 FALSE（或 NA）。因此 ifelse 的输出将是单个元素。如果测试结果为 false，它将从 as.character(floor_date(valDate, sTerm)) 中获取此元素。它只需要一个，因此将采用第一个。然后 mutate 将此单个值回收到列的长度。

如果您希望输出的长度与 valDate 相同，解决方法是重复测试，以便获得所需长度的向量作为测试：

dfDatasetOutput %>%
  mutate(valDate=ifelse(rep(toupper(sTerm)=="CUSTOM",nrow(dfDatasetOutput)), 
                        sCustom, 
                        as.character(floor_date(valDate, sTerm)))) 

为了避免意外使用 ifelse，请考虑使用 if_else 它对对象长度进行检查。