반응형
질문
다음과 같은 코드가 있습니다. 그러나 어떻게 더 좋게 할 수 있을까요? 현재는 중첩된 루프보다는 더 좋다고 생각하지만, 생성기가 목록 표현식에 포함되면 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_date
와end_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()
와 같이요.
반응형
댓글