코드 실행 시간 비교하기

import timeit

# for 루프 코드
for_loop = """
li = [0,1,1,1,0,0,1,1,0,0,1,1,1]
res = []
cnt = 0
for num in li:
    if num:
        cnt += 1
    elif cnt:
        res.append(cnt)
        cnt = 0
if cnt:
    res.append(cnt)
"""

# while 루프 코드
while_loop = """
li = [0, 1, 1, 1, 0, 0, 1, 1, 0, 0, 1, 1, 1]
res = []
i = 0

while i < len(li):
    cnt = 0
    while i < len(li) and li[i]:
        cnt += 1
        i += 1
    if cnt:
        res.append(cnt)
    i += 1
"""

# timeit.timeit() 함수를 사용해 각 코드의 실행 시간을 측정합니다.
# number 매개변수는 코드를 실행할 횟수를 설정합니다.
for_time = timeit.timeit(stmt=for_loop, number=100000)
while_time = timeit.timeit(stmt=while_loop, number=100000)

print(f"for 루프 실행 시간: {for_time}초")
print(f"while 루프 실행 시간: {while_time}초")