# -*- coding: utf-8 -*-
"""
1강 · 5단계 — 함수 (def) : 같은 계산을 이름 하나로
"""
def weekly_return(new, old):
    """주간 수익률 = 새 값 ÷ 옛 값 - 1"""
    return new / old - 1

def moving_average(prices, window):
    """마지막 window 개 가격의 평균"""
    if len(prices) < window:
       return None
    last = prices[-window:]                  # 끝에서 window 개 자르기
    return sum(last) / window

prices = [338.50, 349.54, 343.92, 343.50, 351.66, 349.22]
print(f"이번 주 수익률: {weekly_return(prices[-1], prices[-2]):.2%}")
print(f"5주 이동평균: {moving_average(prices, 5):.2f}")
print("이번 주 가격이 5주 평균보다 높은가?", prices[-1] > moving_average(prices, 5))
print("10주 이동평균:", moving_average(prices, 10))

# ── 해 보기 ──
# ① moving_average(prices, 10) 은? 6개뿐인데 10개 평균 — 결과가 맞나요? (AI 코드에서 자주 보는 함정)
# ② 함수 안에 if len(prices) < window: return None 을 넣어 고쳐 보세요