# -*- coding: utf-8 -*-
"""
3강 · 5단계 — 0교시 복습을 판다스로: 원주가(close)와 수정주가(adjusted close)
0교시에서 'GOOGL 이 -51.3%, -95.2% 폭락'으로 잘못 잡혔던 두 주를 판다스로 3줄 만에 찾는다.
(2부 2-2강 '기업행사와 수정주가' 예고편)
"""
from pathlib import Path
import pandas as pd

g = pd.read_csv(Path(__file__).parent / "data" / "GOOGL_weekly_adjusted.csv", parse_dates=["timestamp"], index_col="timestamp").sort_index()
print("GOOGL 주간:", len(g), "주", g.index[0].date(), "~", g.index[-1].date())

raw = g["close"].pct_change()                 # 원주가 기준 주간 수익률
adj = g["adjusted close"].pct_change()        # 수정주가 기준 주간 수익률

print("\n원주가로 -40% 넘게 '폭락'한 주")
print(raw[raw < -0.40].map("{:+.1%}".format))
print("\n같은 주, 수정주가로는")
print(adj[raw < -0.40].map("{:+.1%}".format))
print(f"\n수정주가로 -40% 넘게 빠진 주: {(adj < -0.40).sum()}개  → 폭락이 아니라 '주식 분할'이었다")

# ── 해 보기 ──
# ① 원주가와 수정주가로 전체 기간 보유 수익률을 각각 구해 보기 (어느 쪽이 진짜 투자 성과인가?)
# ② 배당(dividend amount)이 0보다 큰 주는 몇 번?   →  (g["dividend amount"] > 0).sum()