[Python] 주가데이터 가져오기

Outline

애플의 주가데이터를 가져와서 간단한 차트로 보여줍니다.

Code

필요한 것들을 import 합니다.

import yfinance as yf
import pandas as pd
from pandas_datareader import data as pdr
import datetime as dt

기간은 2021년부터 현재까지로 합니다.

yf.pdr_override()
start_dt = dt.datetime(2021, 1, 1)
end_dt = dt.datetime.now()

데이터를 가져오고, 종가(Close) 대신에 수정종가(Adj Close)를 남깁니다.

df = pdr.get_data_yahoo("AAPL", start_dt, end_dt)
df = df.loc[:, ["Open", "High", "Low", "Adj Close", "Volume"]]
df = df.rename(columns={"Adj Close": "Close"})

데이터를 확인합니다.

Matplotlib를 이용하여 간단히 차트를 그립니다.

import matplotlib.pyplot as plt
plt.figure(figsize = (8, 6))
plt.title("AAPL")
plt.plot(df["Close"])
plt.show()

mplfinance를 이용하여 캔들차트를 그립니다.
상승/하락에 대한 색깔 표현이 나라마다 다르므로 한국에서 사용하는 상승=빨강, 하락=파랑으로 변경합니다.

import mplfinance as mpf
mc = mpf.make_marketcolors(up="r", down="b")
s = mpf.make_mpf_style(base_mpf_style='starsandstripes', marketcolors=mc)
mpf.plot(df, type="candle", style=s, title="AAPL")

Reference

• yfinance: https://github.com/ranaroussi/yfinance
• mplfinance: https://github.com/matplotlib/mplfinance