SeouliteLab

[Python/파이썬] 코드 실행 시간 측정 본문

프로그래밍

[Python/파이썬] 코드 실행 시간 측정

Seoulite Lab 2024. 3. 4. 08:43

파이썬 코드의 실행 시간을 측정하는 것은 코드의 성능을 평가하고 최적화하는 데 도움이 됩니다. 여러 방법을 사용하여 코드 실행 시간을 측정할 수 있습니다. 아래 예제를 통해 각 방법을 살펴보겠습니다.

1. time 모듈 사용

가장 기본적인 방법은 time 모듈을 사용하여 코드 실행 시간을 측정하는 것입니다. 시작 시간과 종료 시간의 차이를 계산하여 측정합니다.

import time

start_time = time.time()

# 실행할 코드 작성
for i in range(1000000):
    pass

end_time = time.time()
execution_time = end_time - start_time
print("실행 시간:", execution_time, "초")

위 코드에서는 time.time() 함수를 사용하여 현재 시간을 가져옵니다. 코드 실행 전과 후에 시간을 가져와서 차이를 계산하여 실행 시간을 측정합니다.

2. timeit 모듈 사용

timeit 모듈은 코드를 여러 번 실행하고 실행 시간의 평균을 계산합니다. 이 모듈을 사용하면 코드 실행 시간을 더 정확하게 측정할 수 있습니다.

import timeit

code_to_measure = '''
# 실행할 코드 작성
for i in range(1000000):
    pass
'''

execution_time = timeit.timeit(code_to_measure, number=1)
print("실행 시간:", execution_time, "초")

위 코드에서는 timeit.timeit() 함수를 사용하여 코드를 실행하고 실행 시간을 측정합니다. number 매개 변수를 사용하여 실행 횟수를 지정할 수 있습니다.