본문 바로가기
Python/Python FAQ

Python 파이썬에서 날짜 범위를 반복하는 것, Iterating through a range of dates in Python

by 베타코드 2023. 10. 9.
반응형

질문


다음과 같은 코드가 있습니다. 그러나 어떻게 더 좋게 할 수 있을까요? 현재는 중첩된 루프보다는 더 좋다고 생각하지만, 생성기가 목록 표현식에 포함되면 Perl-one-linerish해집니다.

day_count = (end_date - start_date).days + 1
for single_date in [d for d in (start_date + timedelta(n) for n in range(day_count)) if d <= end_date]:
    print strftime("%Y-%m-%d", single_date.timetuple())

참고 사항

  • 실제로는 이것을 인쇄하는 데 사용하지 않습니다. 그것은 데모 목적으로만 사용됩니다.
  • start_dateend_date 변수는 타임스탬프가 필요하지 않기 때문에 datetime.date 객체입니다. (보고서를 생성하는 데 사용될 것입니다).

샘플 출력

시작 날짜가 2009-05-30이고 종료 날짜가 2009-06-09인 경우:

2009-05-30
2009-05-31
2009-06-01
2009-06-02
2009-06-03
2009-06-04
2009-06-05
2009-06-06
2009-06-07
2009-06-08
2009-06-09

답변


왜 두 개의 중첩된 반복문이 있는 걸까요? 하나의 반복문으로도 동일한 데이터 목록이 생성되는 것 같은데요:

for single_date in (start_date + timedelta(n) for n in range(day_count)):
    print ...

또한, 목록이 저장되지 않고 단 하나의 생성기만 반복되는 것 같습니다. 생성기 안의 "if"도 필요 없어 보입니다.

결국, 선형 시퀀스는 하나의 반복자만 필요로 해야 합니다.

John Machin과의 토론 이후 업데이트:

아마 가장 우아한 해결책은 날짜 범위를 순회하는 반복문을 완전히 숨기고 추상화하는 생성기 함수를 사용하는 것일 것입니다:

from datetime import date, timedelta

def daterange(start_date, end_date):
    for n in range(int((end_date - start_date).days)):
        yield start_date + timedelta(n)

start_date = date(2013, 1, 1)
end_date = date(2015, 6, 2)
for single_date in daterange(start_date, end_date):
    print(single_date.strftime("%Y-%m-%d"))

NB: 내장된 range() 함수와 일관성을 유지하기 위해 이 반복문은 end_date에 도달하기 전에 멈춥니다. 따라서 포함적인 반복을 위해 다음 날을 사용하면 됩니다. 마치 range()와 같이요.

반응형

댓글