从 Python 到 R 的工作日算法转换？ [关闭]答案

【问题标题】：Weekday alogorithm conversion from Python into R? [closed]从 Python 到 R 的工作日算法转换？ [关闭]
【发布时间】：2019-02-07 07:29:37
【问题描述】：

我已经用 Python 编写了以下函数。对于任何日期，它都会计算出正确的周末或开始日期。我不知道如何用 R 写这个。有人有什么想法吗？

def Week_dates(dt, day, we=False):
    """
    dt is a date
    day is string like 'sat' or 'mon'
    """
    day = day.lower()
    days = ['mon','tue','wed','thu','fri','sat','sun']

    if we:
        day_num = (days.index(day) + 1) % 7
    else:
        day_num = days.index(day)

    day_offset = (dt.weekday() + 7-day_num) % 7
    return dt - timedelta(days=day_offset) + timedelta(days=(6 * we))

【问题讨论】：

请阅读并编辑您的问题：How to make a great R reproducible example?

标签： python r code-translation

【解决方案1】：

使用lubridate包请看如下：

Week_dates <- function(dt, day, we = FALSE) {
  require(lubridate)
  day <- tolower(day)
  days <- c("mon","tue","wed","thu","fri","sat","sun")
  if (we) {
    day_num <- (match(day, days) + 1) %% 7
  } else {
    day_num <- match(day, days) + 1
  }
  day_off_set <- (wday(dt) + 7 - day_num) %% 7
  dt - as.period(day_off_set, unit = "days") + as.period(6 * we, unit = "days")
}

Week_dates(ymd("2012/10/12"), "mon")
# [1] "2012-10-08"

【讨论】：

非常感谢 Artem，非常感谢