python函数判断闰年/天数等

练习一：

写一个判断闰年的函数，参数为年、月、日。若是是闰年，返回True。

目前找到的最简洁的写法，很好用。

#判断闰年
def is_leap_year(year):
    return  (year % 4 == 0 and year % 100 != 0) or year % 400 == 0

引申一下，判断是这一年的第几天。

#判断是这一年的第几天
def getDayInYear(year,month,day):
    month_day = [31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    if is_leap_year(year):
        month_day[1]=29
    return sum(month_day[:month - 1]) + day

print(getDayInYear(2008,1,1))
print(getDayInYear(2008,10,1))
print(getDayInYear(2009,10,1))

当然，python自带datetime模块，我们可以多一种方法去实现。

import datetime        #导入datetime模块
def getDayInYear(year, month, day):
    date = datetime.date(year, month, day)
    return date.strftime('%j')              #返回当天是当年的第几天，范围[001,366]
    
print(getDayInYear(2008,1,1))
print(getDayInYear(2008,10,1))
print(getDayInYear(2009,10,1))

练习二：

有一个文件，文件名为output_1981.10.21.txt 。

下面使用Python：读取文件名中的日期时间信息，并找出这一天是周几。

将文件改名为output_YYYY-MM-DD-W.txt (YYYY:四位的年，MM：两位的月份，DD：两位的日，W：一周的周几，并假设周一为一周第一天)

使用正则和时间函数实现。

import re,datetime
m = re.search('output_(?P<year>d{4}).(?P<month>d{2}).(?P<day>d{2})','output_1981.10.21.txt')
#获取周几，写法一
def getdayinyear(year,month,day):
    date = datetime.date(year,month,day)
    return date.strftime('%w')
W = getdayinyear(1981,10,21)                  #这种写法适用于多个日期

#获取周几，写法二
W = datetime.datetime(int(m.group('year')),int(m.group('month')),int(m.group('day'))).strftime("%w")    #单个日期的写法

#参考
>>> from datetime import datetime
>>> print datetime(1981,10,21).strftime("%w")    #单个日期的写法

filename = 'output_' + m.group('year') + '-' + m.group('month') + '-' + m.group('day') + '-' + W + '.txt'
#可以尝试下用rename方法去写

print (W)
print (filename)