基于 R 中的其他列创建列

Question

我对 R 非常陌生，但我想根据其他 2 列（日期列）在数据框中创建一个新列。我有一个编码为 (1, 2, 3) 的分组变量和 2 个日期列。 以下是我的新列的条件：

• 如果分组列 == 1，那么它应该从 date1 列返回相应的日期（按行）
• 如果分组列 == 2，那么它应该从 date2 返回日期 如果分组列
• == 3，那么它应该带回

我尝试过 case_when 和 if_else 但没有取得任何成功的 2 个日期列之间的最早/第一个日期。任何帮助将不胜感激。

我尝试了 case-when 和 if_else 但出现错误

Answer 1

在组变量为 3 的情况下，我不能 100% 确定这是否是您想要的，但它会为较早的日期添加一天。

df <- data.frame(group = c(1:3),
                 date1 = c(01-02-99, 01-02-99,1-02-99),
                 date2 = c(04-02-99, 04-02-99,04-02-99)
                 )

df$date1 <- as.Date(df$date1)
df$date2 <- as.Date(df$date2)


df <- df %>% mutate(col3 = case_when(
    group == 1 ~ date1,
    group == 2 ~ date2,
    group == 3 & date1 > date2 ~ date2+1,
    group == 3 & date2 > date1 ~ date1+1
))

Answer 2

这是 case_when 的一种方法。

set.seed(2022)

d <- seq(as.Date("2022-01-01"), Sys.Date(), by = "day")
df1 <- data.frame(
  group = sample(3, 20, TRUE),
  date1 = sample(d, 20),
  date2 = sample(d, 20)
)

suppressPackageStartupMessages(library(dplyr))

df1 %>%
  mutate(newdate = case_when(
    group == 1 ~ date1,
    group == 2 ~ date2,
    TRUE ~ pmin(date1, date2)
  )) %>% head
#>   group      date1      date2    newdate
#> 1     3 2022-01-29 2022-02-25 2022-01-29
#> 2     2 2022-01-12 2022-01-01 2022-01-01
#> 3     3 2022-03-20 2022-01-19 2022-01-19
#> 4     3 2022-03-01 2022-03-01 2022-03-01
#> 5     2 2022-02-13 2022-02-19 2022-02-19
#> 6     3 2022-03-11 2022-02-10 2022-02-10

^{由 reprex 软件包 (v2.0.1) 创建于 2022 年 3 月 24 日}