Python如何在带条件的列表理解中使用多个for循环

问题描述

我仍在处理以下代码，代码工作正常。我正在尝试减少代码行。

import calendar as c
def solve(first,last):
    weekends = []
    # x = [weekends.append(m) if c.weekday(y,m,1) == 4 and c.weekday(y,31) == 6 else 0 for m in [1,3,5,7,8,10,12] for y in range(first,last+1)]
    for y in range(first,last+1):
        for m in [1,12]:
            if c.weekday(y,31) == 6:
                weekends.append(m)
    return c.month_abbr[weekends[0]],c.month_abbr[weekends[len(weekends)-1]],len(weekends)

在需要时：solve(2016,2020)

此代码返回2016年的第一个月，其中有5个星期五，星期六，星期日； 2020年最后一个月的情况相同，有多少个月可以满足此条件。

所以OUTPUT是：('Jan','May',5)

x变量的注释部分是我尝试的，它返回0和None（是else语句的原因）

解决方法

x = ...中语句的顺序有些混乱；您的if应该过滤要包括的值，而不是要使用的两个替代值中的哪一个。而且：不要在列表理解中使用append附加到另一个列表！相反，列表理解本身应该是您的结果。

def solve(first,last):
    weekends = [c.month_abbr[m] for y in range(first,last+1)
                                for m in [1,3,5,7,8,10,12]
                                if c.weekday(y,m,1) == 4]
    return weekends[0],weekends[-1],len(weekends)

我修正的一些小问题：

• 直接在列表中获取month_abbr，而不是在末尾添加两次
• -1本身是有效索引
• 两个工作日检查都是多余的