[Python] timeit 활용해서 코드 실행 속도 비교하기

안녕하세요

오늘은 timeit 모듈 사용해서, 코드 실행 속도를 비교해보도록 하겠습니다.

 

timeit은 파이썬 내장 모듈로,
코드 조각의 실행 속도를 정밀하게 측정하는 도구입니다.

 

전체 코드입니다

 

main.py

import timeit

# 1. 테스트할 코드 조각 정의
code_list = "[x**2 for x in range(1000)]"  # 리스트 컴프리헨션
code_loop = """
result = []
for x in range(1000):
    result.append(x**2)
"""

# 2. 각 코드를 10,000번 실행하여 평균 시간 측정
time_list = timeit.timeit(stmt=code_list, number=10000)
time_loop = timeit.timeit(stmt=code_loop, number=10000)

print("=== 실행 속도 비교 결과 ===")
print(f"리스트 컴프리헨션: {time_list:.5f}초")
print(f"for문: {time_loop:.5f}초")

# 3. 어느 쪽이 더 빠른지 출력
if time_list < time_loop:
    print("\n 리스트 컴프리헨션이 더 빠릅니다!")
else:
    print("\n for문이 더 빠릅니다!")

 

실행 결과입니다.

 

 

timeit은 여러 번 실행한 평균 속도를 자동으로 계산해주기 때문에,

이를 활용해 노이즈 없는 성능 비교에 활용할 수 있습니다.