如何用列表推导式改写获取指定日期区间月最后一天的函数？

首先明确：列表推导式的条件判断逻辑和for循环完全一致，你遇到的空列表问题，通常出在循环迭代范围、变量生成逻辑或条件放置位置上，而非逻辑本身的差异。

先假设你的原嵌套循环逻辑类似以下可正常运行的版本：

from datetime import datetime, timedelta

def get_month_ends_loop(start: datetime, end: datetime):
    result = []
    current_year, current_month = start.year, start.month
    end_year, end_month = end.year, end.month
    
    while (current_year, current_month) <= (end_year, end_month):
        # 计算当月最后一天：下个月首日减1天
        next_year = current_year if current_month < 12 else current_year + 1
        next_month = current_month + 1 if current_month < 12 else 1
        month_end = datetime(next_year, next_month, 1) - timedelta(days=1)
        # 过滤超出起止范围的日期
        if start <= month_end <= end:
            result.append(month_end)
        # 推进到下一个月
        current_month += 1
        if current_month > 12:
            current_month = 1
            current_year += 1
    return result

正确的列表推导式实现 #

要转成列表推导式，核心是先生成所有需要处理的年月组合，再对每个年月计算最后一天并过滤：

简洁版（抽离日期计算逻辑）

from datetime import datetime, timedelta

def get_last_day(year: int, month: int) -> datetime:
    next_year = year if month < 12 else year + 1
    next_month = month + 1 if month < 12 else 1
    return datetime(next_year, next_month, 1) - timedelta(days=1)

def get_month_ends_listcomp(start: datetime, end: datetime):
    # 生成起止日期之间的所有年月对
    year_months = []
    current_year, current_month = start.year, start.month
    while (current_year, current_month) <= (end.year, end.month):
        year_months.append((current_year, current_month))
        current_month += 1
        if current_month > 12:
            current_month = 1
            current_year += 1
    
    # 列表推导式：遍历年月→计算最后一天→过滤有效日期
    return [
        get_last_day(y, m)
        for y, m in year_months
        if start <= get_last_day(y, m) <= end
    ]

一体化版本（不抽离函数）

from datetime import datetime, timedelta

def get_month_ends_listcomp(start: datetime, end: datetime):
    year_months = []
    current_year, current_month = start.year, start.month
    while (current_year, current_month) <= (end.year, end.month):
        year_months.append((current_year, current_month))
        current_month += 1
        if current_month > 12:
            current_month = 1
            current_year += 1
    
    return [
        datetime(y, m+1, 1) - timedelta(days=1) if m < 12 else datetime(y+1, 1, 1) - timedelta(days=1)
        for y, m in year_months
        if start <= (datetime(y, m+1, 1) - timedelta(days=1) if m < 12 else datetime(y+1, 1, 1) - timedelta(days=1)) <= end
    ]

你之前得到空列表的常见原因 #

• 年月范围生成错误：生成的year_months为空，或未覆盖起止日期之间的所有月份；
• 条件判断逻辑错误：比如将start <= month_end <= end写反，导致所有日期都被过滤；
• 最后一天计算错误：比如误算成当月第一天而非最后一天，导致条件判断不满足。

列表推导式与for循环的条件差异 #

两者的条件判断逻辑完全等价，只是写法顺序不同：

• for循环是「计算元素→判断条件→满足则添加」；
• 列表推导式是「遍历迭代器→计算元素→条件过滤→保留满足的元素」。
  只要条件表达式与原循环完全一致，结果就会完全相同。

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