[스파르타코딩] 파이선 지표분석 4주차

드디어 수업이 끝났다. 이제 뭐하지?

#1.여러기업의 상관계수를 보는 작업
#1.1 판다스 불러오기

import pandas as pd
import pandas_datareader as pdr
from datetime import datetime

#1.2 주가데이터 불러오기

for company in companies:
    company_code = get_code(code_result, company)
    stock_df = pdr.get_data_yahoo(company_code, start, end)
    stocks_of_companies = stocks_of_companies.join(pd.DataFrame(stock_df['Close']).rename(columns={'Close':company}),on = 'Date')

#1.3 라이브러리에서 파이플롯고 씨본 추가

import matplotlib.pyplot as plt
import seaborn as sns

#1.4 히트맵 그리기

plt.figure(figsize=(10,10))
sns.heatmap(data = corr_data, annot=True, fmt = '.2f', linewidths = .5, cmap = 'Blues')
plt.show()