日期天数转换（python3）

【问题描述】

编写一个程序，用户输入日期，计算该日期是这一年的第几天。

【输入形式】

用户在第一行输入一个日期（年 月 日,中间以空格分割） 

【输出形式】

程序在下一行输出一个整数

【样例输入】

2006 2 21

【样例输出】

52

【样例说明】

用户以年月日的格式输入，中间以空格分割，程序计算出该天是输入年份的第几天并输出该天数。另外还需要判断这一年是否为闰年。

【代码】

def f(n):
    if n % 4 == 0 and n % 100 != 0 or n % 400 == 0:
        return True
    else:
        return False


if __name__ == '__main__':
    s = input().split(' ')
    l = [int(x) for x in s]
    month = [0, 31, 0, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    if f(l[0]):
        month[2] = 29
    else:
        month[2] = 28
    sum = 0
    for i in range(1, l[1]):
        sum += month[i]
    sum += l[2]
    print("%d" % sum)