R语言ifelse多条件生成季节变量时Winter分类失效求助

问题描述 #

使用以下代码为月份分配季节时，Winter分类完全未生效：

data$SEASON = ifelse(data$Month > 8, "Fall", 
                            ifelse(data$Month > 10, "Winter",
                            ifelse(data$Month < 6, "Spring","Summer")))

运行后无报错，但结果不符合预期：

• 1-5月被标记为Spring（实际1-3月应为Winter）
• 9-12月被标记为Fall（实际11-12月应为Winter）
• 仅Summer（6-9月）分配正确

期望实现：1、2、3、11、12月标记为Winter，4-5月为Spring，6-9月为Summer，10月为Fall。

原始数据示例：

Year    Month   Day 
  2018    1       1 
  2018    1       2 
  2018    2       3 
  2018    3       3 
  2018    9       1
  2018    8       4
  2018    7       5
  2018    6       2
  2018    5       4 
  2018    6       7  
  2018    10      4
  2018    11      5
  2018    12      8  
  2018    3       9
  2018    4       10 

问题分析 #

ifelse是从上到下的顺序判断，代码逻辑存在优先级错误：

1. 第一个判断Month >8会直接捕获9-12月，标记为Fall，后续的Month>10判断根本不会执行
2. 第二个判断Month <6会捕获1-5月，标记为Spring，1-3月无法进入后续逻辑

解决方案 #

调整判断顺序，先处理Winter的区间，再依次处理其他季节：

方案1：使用%in%指定精确月份 #

data$SEASON = ifelse(data$Month %in% c(1,2,3,11,12), "Winter",
                     ifelse(data$Month %in% c(4,5), "Spring",
                            ifelse(data$Month %in% c(6,7,8,9), "Summer", "Fall")))

方案2：使用区间判断更简洁 #

data$SEASON = ifelse((data$Month >=1 & data$Month <=3) | (data$Month >=11 & data$Month <=12), "Winter",
                     ifelse(data$Month >=4 & data$Month <=5, "Spring",
                            ifelse(data$Month >=6 & data$Month <=9, "Summer", "Fall")))

验证结果 #

用原始数据测试后，结果符合预期：

• 1、2、3、11、12月 → Winter
• 4、5月 → Spring
• 6-9月 → Summer
• 10月 → Fall

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