请求协助：将R语言日期处理代码转换为自定义函数

现有一段可正常运行的R代码，用于补全不完整的日期字符串，并将其转换为与基准日期1899-12-30的间隔整数。尝试将这段代码封装为接收数据框、列名、行号作为参数的自定义函数时出现错误，以下是问题分析与修复方案：

原可运行代码 #

library(tidyverse)
library(admiral)
library(lubridate)

date_df <- tribble(
  ~id, ~year,
  1, "2020-01-01",
  2, "2021-06-15",
  3, "2023-12-31",
  4, "1999",
  5, "1978",
  6, "2001-10-07",
)

# 运行结果为：36525
as.integer(as_date(admiral::impute_dtc_dt(dtc = date_df$year[4], highest_imputation = "M", date_imputation = "06-15")) - as_date("1899-12-30"))

用户编写的错误函数 #

# 无法正常运行的自定义函数
impute_year <- function(date_df, year, rownum) {
  as.integer(as_date(admiral::impute_dtc_dt(dtc = date_df$!!year[!!rownum], highest_imputation = "M", date_imputation = "06-15")) - as_date("1899-12-30"))
}

问题原因 #

函数中date_df$!!year[!!rownum]的写法存在两个核心问题：

1. $运算符不支持tidyeval的!!（非标准求值）语法，无法通过!!year动态解析列名；
2. 当year作为字符串参数传入时，$会将year作为字面量而非变量解析，导致无法正确提取目标列。

修复后的函数 #

方案1：支持字符串列名参数 #

直接使用[[运算符提取列，这是R中动态获取数据框列的标准方式：

impute_year <- function(date_df, year, rownum) {
  # 提取指定行的日期字符串
  target_dtc <- date_df[[year]][rownum]
  # 执行日期补全与转换逻辑
  as.integer(
    as_date(admiral::impute_dtc_dt(dtc = target_dtc, highest_imputation = "M", date_imputation = "06-15")) - 
    as_date("1899-12-30")
  )
}

# 测试调用，返回结果36525
impute_year(date_df, "year", 4)

方案2：同时支持裸变量与字符串列名 #

如果需要支持像impute_year(date_df, year, 4)这样的裸变量列名调用，可以结合tidyeval语法处理列名：

impute_year <- function(date_df, year, rownum) {
  # 将输入的列名（裸变量或字符串）转换为字符串
  year_col <- rlang::as_name(enquo(year))
  target_dtc <- date_df[[year_col]][rownum]
  
  as.integer(
    as_date(admiral::impute_dtc_dt(dtc = target_dtc, highest_imputation = "M", date_imputation = "06-15")) - 
    as_date("1899-12-30")
  )
}

# 两种调用方式都可行
impute_year(date_df, "year", 4)
impute_year(date_df, year, 4)

内容的提问来源于stack exchange，提问作者microbe