如何简化日期规则?多区间日期处理规则优化咨询
简化日期规则的方案
先指出你原代码里的明显错误:第二个if的结束条件错用了rule1_1993_end,应该是rule2_1993_end;第三个if的结束条件也错用了rule1_1993_end,应该是rule3_1993_end,这些笔误会导致逻辑判断完全偏离预期。
针对你的需求,核心是按年份统一复用规则,不用给每个年份单独定义一堆重复变量,以下是两种简洁的实现思路:
思路1:基于日期月份区间直接判断
利用日期的月份规律,动态生成当年的规则区间,避免冗余变量定义:
from datetime import datetime date_str = "1993-03-04" # 待处理日期 dt = datetime.strptime(date_str, "%Y-%m-%d") # 动态生成当年的各规则区间 jan_period = (datetime(dt.year, 1, 1), datetime(dt.year, 1, 31)) feb_may_period = (datetime(dt.year, 2, 1), datetime(dt.year, 5, 31)) jun_dec_period = (datetime(dt.year, 6, 1), datetime(dt.year, 12, 1)) if jan_period[0] <= dt <= jan_period[1]: # 1月区间:替换年份为1995,保留原月日 new_date = f"1995-{dt.month:02d}-{dt.day:02d}" elif feb_may_period[0] <= dt <= feb_may_period[1]: # 2-5月区间:固定返回2000-01-01 new_date = "2000-01-01" elif jun_dec_period[0] <= dt <= jun_dec_period[1]: # 6-12月1日区间:保留原日期 new_date = date_str else: # 未覆盖的日期(如12月2日及以后),可自定义处理逻辑 new_date = date_str print(new_date)
思路2:用规则列表统一管理(便于扩展)
如果之后需要添加更多年份或规则,可以把所有规则整理成列表,每个规则包含判断条件和处理逻辑,循环匹配即可:
from datetime import datetime def process_date(date_str): dt = datetime.strptime(date_str, "%Y-%m-%d") # 规则列表:(区间判断函数, 结果生成函数) rules = [ # 当年1月1日-1月31日 (lambda d: datetime(d.year, 1, 1) <= d <= datetime(d.year, 1, 31), lambda d: f"1995-{d.month:02d}-{d.day:02d}"), # 当年2月1日-5月31日 (lambda d: datetime(d.year, 2, 1) <= d <= datetime(d.year, 5, 31), lambda d: "2000-01-01"), # 当年6月1日-12月1日 (lambda d: datetime(d.year, 6, 1) <= d <= datetime(d.year, 12, 1), lambda d: date_str) ] # 遍历规则,执行第一个匹配的逻辑 for condition, handler in rules: if condition(dt): return handler(dt) # 默认返回原日期 return date_str # 测试示例 print(process_date("1993-01-12")) # 输出:1995-01-12 print(process_date("1993-03-04")) # 输出:2000-01-01 print(process_date("1993-07-01")) # 输出:1993-07-01 print(process_date("1994-12-01")) # 输出:1994-12-01
这两种方式都彻底消除了原代码中按年份重复定义变量的冗余问题,逻辑更清晰,也方便后续维护和扩展。
内容的提问来源于stack exchange,提问作者reynold02
相关产品推荐
相关产品推荐

