日付をイテレート

日別に整理されたデータがある中で、一定期間のデータだけ分析を行いたいことはよくある。

期間を文字列で指定すると、1日ずつ順番に日付を返すジェネレータがあると便利。

• 入力文字列の指定方法
  • <Date> ::= “YYYYMMDD”形式の日付文字列
  • <Dates>::= <Date> | <Date>-<Date> | <Dates>,<Dates>
  • <Dates>を指定する
• ハイフン区切り「YYYYMMDD-YYYYMMDD」は、期間を表す
• カンマ区切り「YYYYMMDD,YYYYMMDD」は、別々の2日を表す

(例)20150101-20150331,20150701,20150801
⇒ 2015/01/01～2015/03/31の各日、2015/07/01、2015/08/01

安心と信頼の、ノーエラーチェック

from datetime import datetime
from dateutil.relativedelta import relativedelta

# 補助関数
def _generate_between_date(start_date_str, stop_date_str):
    start_date = datetime.strptime(start_date_str, '%Y%m%d')
    stop_date = datetime.strptime(stop_date_str, '%Y%m%d')
    while start_date <= stop_date:
        yield start_date.strftime('%Y%m%d')
        start_date += relativedelta(days=1)

# こちらを使う
def date_sequence_generator(dates_str):
    for date_str in dates_str.split(','):
        if '-' in date_str:
            for date in _generate_between_date(*date_str.split('-')):
                yield date
        else:
            yield date_str

from datetime import datetime
from dateutil.relativedelta import relativedelta

def _generate_between_month(start_month_str, stop_month_str):
    start_month = datetime.strptime(start_month_str, '%Y%m')
    stop_month = datetime.strptime(stop_month_str, '%Y%m')
    while start_month <= stop_month:
        yield start_month.strftime('%Y%m')
        start_month += relativedelta(months=1)

def month_sequence_generator(months_str):
    for month_str in months_str.split(','):
        if '-' in month_str:
            for month in _generate_between_month(*month_str.split('-')):
                yield month
        else:
            yield month_str