更新日期并计数(如果符合条件)

Question

我写了以下代码：

daysInMonths = [31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
startDate = {'day': 1, 'month': 1, 'year': 1901, 'dayOfTheWeek': 3}
endDate = {'day': 31, 'month': 12, 'year': 2000, 'dayOfTheWeek': 1}
counter = 0

def update_date():
    startDate['day'] += 1
    startDate['dayOfTheWeek'] += 1
    if startDate['dayOfTheWeek'] > 7:
        startDate['dayOfTheWeek'] = 1
    if startDate['day'] > daysInMonths[startDate['month'] - 1]:
        startDate['day'] = 1
        startDate['month'] += 1
    if startDate['month'] > 12:
        startDate['month'] = 1
        startDate['year'] += 1
    if startDate['year'] % 4 == 0 and not startDate['year'] % 100 == 0:
        daysInMonths[2] = 29

def compare():
    if startDate['day'] != endDate['day']:
        return True
    if startDate['month'] != endDate['month']:
        return True
    if startDate['year'] != endDate['year']:
        return True
    return False
while compare():
    if startDate['day'] == startDate['dayOfTheWeek'] == 1:
        counter += 1
        print(counter)
    print(startDate)
    update_date()

所以我试图计算每月第一天的星期日，但我得到一个糟糕的结果 (173(，任何建议我的代码出错了？

Answer 1

您的整个代码可以表达为：

import calendar
import itertools
first_of_months = itertools.product(range(1901, 2001),
                                    range(1,13),
                                    itertools.repeat(1)):
result = sum(1 for y,m,d in first_of_months
             if calendar.weekday(y,m,d) == calendar.SUNDAY)

或等效于：

from calendar import weekday, SUNDAY
total = 0
for y in range(1901, 2001):
    for m in range(1, 13):
        d = 1
        if weekday(y, m, d) == SUNDAY:
            total += 1

Answer 2

一个更短的解决方案(更多的是出于好奇而不是必要(：

import calendar
sum(calendar.monthrange(year,month)[0]==calendar.SUNDAY 
    for year in range(1901,2001) 
    for month in range(1,13))