R:将日期解析为年/季度

时间:2018-09-01 21:43:17

标签: r function functional-programming zoo lubridate

我有一个看起来像这样的数据框

Classes ‘tbl_df’, ‘tbl’ and 'data.frame':   344479 obs. of  6 variables:
 $ REGION        : chr  "NSW1" "NSW1" "NSW1" "NSW1" ...
 $ SETTLEMENTDATE: POSIXct, format: "1998-12-07 02:00:00" "1998-12-07 
02:30:00" "1998-12-07 03:00:00" "1998-12-07 03:30:00" ...
 $ TOTALDEMAND   : num  3294 5337 5296 5266 5330 ...
 $ RRP           : num  8.01 11.16 13.52 12.52 13.01 ...
 $ PERIODTYPE    : chr  "TRADE" "TRADE" "TRADE" "TRADE" ...
 $ month         : num  12 12 12 12 12 12 12 12 12 12 ...

我正在尝试通过从year_quarter变量中提取变量来创建一个2014-Q1变量,该变量是一个字符串,其格式为:SETTLEMENTDATE(表示年/季度)。

有许多通过zoolubridate解决此问题的方法,但我希望有人可以告诉我为什么我的函数不起作用起作用:

quarter_fun <- function(df){
    df$quarter <- NA
    if (df$month <= 3){
    df$quarter <- paste(format(df$SETTLEMENTDATE, format = "%Y")[1], 
"Q1", sep="-")
    } else if (df$month >= 4 & df$month <= 6){ 
      df$quarter <- paste( format(df$SETTLEMENTDATE, format = "%Y")[1], 
"Q2", sep="-")            
    } else if (df$month >= 7 & df$month <= 9){ 
      df$quarter <- paste(format(df$SETTLEMENTDATE, format = "%Y")[1], 
"Q3", sep="-")
    } else if (df$month == 10){ 
    df$quarter <- paste(format(df$SETTLEMENTDATE, format = "%Y")[1], 
"Q4", sep="-")
    }

}

我收到此错误消息:

the condition has length > 1 and only the first element will be usedthe 
condition has length > 1 and only the first element will be usedthe 
condition has length > 1 and only the first element will be usedthe 
condition has length > 1 and only the first element will be used

任何帮助将不胜感激-同样,这与寻找解决当前任务的方法无关,而在于了解为什么我的尝试不起作用,因为我的某处显然存在错误的假设(或多个假设)一路走来。

谢谢!

1 个答案:

答案 0 :(得分:1)

您的解决方案忽略了df$month是向量的事实,并且if需要计算为单个true / false值。您的比较产生了真/假值的逻辑向量。因此,警告消息“将仅使用第一个元素”。

相反,请考虑使用cut重新标记数字月份:

numeric.months <- 1:12
quarters <- cut(numeric.months, seq(0, 12, 3), labels = paste0('Q', 1:4), include.lowest = T)

 [1] Q1 Q1 Q1 Q2 Q2 Q2 Q3 Q3 Q3 Q4 Q4 Q4
Levels: Q1 Q2 Q3 Q4