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)