計算當前月份的日期范圍

問題

你的代碼需要在當前月份中循環每一天，想找到一個計算這個日期范圍的高效方法。

解法

在這樣的日期上循環并需要事先構造一個包含所有日期的列表。你可以先計算出開始日期和結束日期，然后在你步進的時候使用 datetime.timedelta 對象遞增這個日期變量即可。

下面是一個接受任意 datetime 對象并返回一個由當前月份開始日和下個月開始日組成的元組對象。

from datetime import datetime, date, timedelta import calendar def get_month_range(start_date=None):if start_date is None:start_date = date.today().replace(day=1)_, days_in_month = calendar.monthrange(start_date.year, start_date.month)end_date = start_date + timedelta(days=days_in_month)return (start_date, end_date)

有了這個就可以很容易的在返回的日期范圍上面做循環操作了：

>>> a_day = timedelta(days=1) >>> first_day, last_day = get_month_range() >>> while first_day < last_day: ... print(first_day) ... first_day += a_day ... 2012-08-01 2012-08-02 2012-08-03 2012-08-04 2012-08-05 2012-08-06 2012-08-07 2012-08-08 2012-08-09 #... and so on...

討論

上面的代碼先計算出一個對應月份第一天的日期。一個快速的方法就是使用 date 或 datetime 對象的 replace() 方法簡單的將 days 屬性設置成 1 即可。 replace()方法一個好處就是它會創建和你開始傳入對象類型相同的對象。所以，如果輸入參數是一個 date 實例，那么結果也是一個 date 實例。同樣的，如果輸入是一個 datetime實例，那么你得到的就是一個 datetime 實例。

然后，使用 calendar.monthrange() 函數來找出該月的總天數。任何時候只要你想獲得日歷信息，那么 calendar 模塊就非常有用了。 monthrange() 函數會返回包含星期和該月天數的元組。

一旦該月的天數已知了，那么結束日期就可以通過在開始日期上面加上這個天數獲得。有個需要注意的是結束日期并不包含在這個日期范圍內 (事實上它是下個月的開始日期)。這個和 Python 的 slice 與 range 操作行為保持一致，同樣也不包含結尾。

為了在日期范圍上循環，要使用到標準的數學和比較操作。比如，可以利用 timedelta 實例來遞增日期，小于號< 用來檢查一個日期是否在結束日期之前。

理想情況下，如果能為日期迭代創建一個同內置的 range() 函數一樣的函數就好了。幸運的是，可以使用一個生成器來很容易的實現這個目標：

def date_range(start, stop, step):while start < stop:yield startstart += step

下面是使用這個生成器的例子：

>>> for d in date_range(datetime(2012, 9, 1), datetime(2012,10,1), timedelta(hours=6)): ... print(d) ... 2012-09-01 00:00:00 2012-09-01 06:00:00 2012-09-01 12:00:00 2012-09-01 18:00:00 2012-09-02 00:00:00 2012-09-02 06:00:00 ... >>>

這種實現之所以這么簡單，還得歸功于 Python 中的日期和時間能夠使用標準的數學和比較操作符來進行運算。

總結

以上是生活随笔為你收集整理的《Python Cookbook 3rd》笔记（3.14）：计算当前月份的日期范围的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。