7.3.2 주가지수 데이터 가져오기
S&P500지수(티커는 ‘SPY’)를 가져오자. web.DataReader를 사용해 주가지수 데이터를 가져온 후 데이터프레임인 df 변수에 저장한다. 혹시 이런저런 이유로 빠진 데이터가 있을 것을 대비해 미리 dropna()를 사용해 제거한다. df[ [ 'Open', 'High', 'Low', 'Close'] ]를 사용해 주가지수 데이터 중 모델과 전략에 사용할 시가, 고가, 저가, 종가만 저장한다.
df = web.DataReader( 'SPY', data_source='yahoo', start='2015-01-01', end='2020-01-01' )
df = df[ [ 'Open', 'High', 'Low', 'Close' ] ]
또는 다음과 같이 가져옴과 동시에 시가, 고가, 저가, 종가만 저장하는 한 줄의 코드로 작성할 수 있다.
df = web.DataReader( 'SPY', data_source='yahoo', start='2016-01-01', end='2020-01-01' )[ [ 'Open', 'High', 'Low', 'Close' ] ]
좀 더 수정해서 DataReader() 함수에 전달하는 매개변수(symbol, source, start, end)도 분리해 다음과 같이 좀 더 보기 편하고 수정하기 쉬운 코드로 만들 수 있다.
symbol, source, start, end = 'SPY', 'yahoo', '2016-01-01', '2020-01-01'
df = web.DataReader( symbol, source, start, end )[ [ 'Open', 'High', 'Low', 'Close' ]