r - 如何防止 ifelse() 将 Date 对象转换为数字对象

Question

我正在使用函数 ifelse() 来操作日期向量。我预计结果是 Date 类，但很惊讶地得到了 numeric 向量。这是一个例子:

dates <- as.Date(c('2011-01-01', '2011-01-02', '2011-01-03', '2011-01-04', '2011-01-05'))
dates <- ifelse(dates == '2011-01-01', dates - 1, dates)
str(dates)

这尤其令人惊讶，因为在整个向量上执行操作会返回一个 Date 对象。

dates <- as.Date(c('2011-01-01', '2011-01-02', '2011-01-03', '2011-01-04','2011-01-05'))
dates <- dates - 1
str(dates)

我应该使用其他函数来操作 Date 向量吗？如果有的话，有什么功能？如果不是，我如何强制ifelse返回与输入相同类型的向量？

ifelse 的帮助页面表明这是一个功能，而不是一个错误，但我仍在努力寻找对我发现的令人惊讶的行为的解释。

Answer 1

您可以使用data.table::fifelse (data.table >= 1.12.3) 或dplyr::if_else。

<小时/>

data.table::fifelse

Unlike ifelse, fifelse preserves the type and class of the inputs.

library(data.table)
dates <- fifelse(dates == '2011-01-01', dates - 1, dates)
str(dates)
# Date[1:5], format: "2010-12-31" "2011-01-02" "2011-01-03" "2011-01-04" "2011-01-05"

<小时/>

dplyr::if_else

来自dplyr 0.5.0 release notes :

[if_else] have stricter semantics that ifelse(): the true and false arguments must be the same type. This gives a less surprising return type, and preserves S3 vectors like dates" .

library(dplyr)
dates <- if_else(dates == '2011-01-01', dates - 1, dates)
str(dates)
# Date[1:5], format: "2010-12-31" "2011-01-02" "2011-01-03" "2011-01-04" "2011-01-05"