R语言提取quarterly_values报错及替代方案咨询

问题根源 #

你遇到的报错是因为seq()函数的to参数传入了向量而非单个标量，通常出现在试图直接通过位置索引匹配季度月份时，错误地将多个位置值传给了seq()的终止参数。

示例环境与错误复现 #

先模拟2014-2023的月度数据集，复现你的错误操作：

library(lubridate)

# 生成10年月度数据
df <- data.frame(
  date = seq.Date(as.Date("2014-01-01"), as.Date("2023-12-01"), by = "month"),
  mont_vals = rnorm(120, 100, 10)
)

# 错误写法：试图通过seq匹配季度第三个月位置，导致to为向量
df$quarterly_vals <- NA
df$quarterly_vals[seq(1, which(month(df$date) %in% c(3,6,9,12)), by=3)] <- df$mont_vals[which(month(df$date) %in% c(3,6,9,12))]

报错信息：

Error in seq.default(1, which(month(df$date) %in% c(3, 6, 9, 12)), by = 3) : 
  'to' must be of length 1

可行解决方案 #

以下提供两种高效的替代方法，均能实现“每个季度第一行赋值该季度第三个月的mont_vals，其余行设为NA”的需求。

方案1：使用dplyr分组处理

借助dplyr的分组功能，按季度分组后定位组内第一行和最后一行（即季度第三个月）：

library(dplyr)
library(lubridate)

df_processed <- df %>%
  # 生成季度起始日期作为分组标识
  mutate(year_quarter = floor_date(date, "quarter")) %>%
  group_by(year_quarter) %>%
  # 组内第一行赋值该季度最后一个月的mont_vals，其余为NA
  mutate(quarterly_vals = ifelse(row_number() == 1, last(mont_vals), NA)) %>%
  ungroup() %>%
  select(-year_quarter) # 移除辅助分组列

方案2：Base R原生实现

无需加载额外包，通过分组统计和索引匹配完成赋值：

# 生成年-季度字符串作为分组键
df$year_quarter <- format(df$date, "%Y-Q%q")

# 获取每个季度第一行的索引，以及对应季度最后一个月的mont_vals值
quarter_first_rows <- !duplicated(df$year_quarter)
quarter_last_vals <- tapply(df$mont_vals, df$year_quarter, tail, n = 1)

# 赋值季度值
df$quarterly_vals <- NA
df$quarterly_vals[quarter_first_rows] <- quarter_last_vals[df$year_quarter[quarter_first_rows]]

# 移除辅助列
df$year_quarter <- NULL

预期输出示例 #

执行上述任一方案后，数据集前6行的结果如下：

date mont_vals quarterly_vals
1 2014-01-01  95.34858       104.7624
2 2014-02-01 102.19404             NA
3 2014-03-01 104.76238             NA
4 2014-04-01  98.87761        97.6041
5 2014-05-01 101.56377             NA
6 2014-06-01  97.60410             NA

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