해커를 꿈꾸는 개발자

파이썬 주식 / yfinance를 이용하여 삼성전자 주식 정보 가져와 그래프로 표시하기

필요 라이브러리

먼저 주식 정보를 받아오기 위해 야후 파이낸스에서 제공하는 yfinace 라이브러리를 사용합니다. 

받아온 정보를 데이터프레임으로 만들기 위해 panda_datareader 라이브러리를 사용합니다. 

그리고 그래프를 그려주기 위해 matplotlib.pyplot 라이브러리를 사용합니다 

from pandas_datareader import data as pdr
import yfinance as yf
import matplotlib.pyplot as plt

주식 시세 구하기

주식 시세를 구하기 위해 get_data_yahoo() 함수를 이용합니다. 

이 함수에는 주식 종목(코드)와 조회 기간을 전달인자로 넘겨주게 됩니다. 국내 주식은 주식 종목 코드가 보통 6자리 숫자 코드로 표현됩니다. 

get_data_yahoo(주식 종목(코드), 조회 기간)

아래는 삼성전자 코드인 '005930.KS'와 시작 날짜인 '2020-07-01'를 전달 인자로 넣어 가져온 데이터 입니다.

가져온 데이터를 분석해보면, index에는 시작 날짜부터 오늘 날짜까지의 날짜 정보가 저장됩니다. 

Columns에는 6가지의 데이터를 가져오게 됩니다. 

여기서 Close 열은 종가 데이터를 의미합니다. 

이 데이터를 가지고 그래프에 그려보겠습니다. 

간단하게, x 축은 index (날짜 정보)로 설정하고, y 축은 Close(종가 정보)로 설정하여, 그래프를 출력하도록 하였습니다.

plt.plot(samsung.index, samsung.Close, 'b', label='Samsung Electronics')
plt.show()

전체 코드

from pandas_datareader import data as pdr
import yfinance as yf
import matplotlib.pyplot as plt

yf.pdr_override()

samsung = pdr.get_data_yahoo('005930.KS', start='2020-07-01')
print(samsung)

print(samsung.index)
print(samsung.columns)

plt.plot(samsung.index, samsung.Close, 'b', label='Samsung Electronics')
plt.show()