[Python] 그래프 4개를 한번에 그리는 방법

'''
pip install pandas,beautifulsoup4,finance-datareader,matplotlib -y
'''

import pandas as pd
import matplotlib.pyplot as plt

import FinanceDataReader as fdr

from datetime import datetime
from dateutil.relativedelta import relativedelta

def get_stock_code(name):
    '''이름을 입력하면 코드를 리턴'''

    df = fdr.StockListing('KRX')
    stock_code = df[df['Name'] == name]['Code'].to_string(index = False)
    return stock_code

if __name__ == '__main__':

    before_standard = (datetime.now() - relativedelta(years = 2)).strftime('%Y-%m-%d')

    df_exchange_rate    = fdr.DataReader(symbol = 'USD/KRW', start = before_standard)[['Close']]
    df_kospi            = fdr.DataReader(symbol = 'KS11', start = before_standard)[['Close']]
    df_samsung_elec     = fdr.DataReader(symbol = get_stock_code('삼성전자'), start = before_standard)[['Close']]
    df_samsung_pre_elec = fdr.DataReader(symbol = get_stock_code('삼성전자우'), start = before_standard)[['Close']]
    df_sk_hynix         = fdr.DataReader(symbol = get_stock_code('SK하이닉스'), start = before_standard)[['Close']]
    df_db_hitek         = fdr.DataReader(symbol = get_stock_code('DB하이텍'), start = before_standard)[['Close']]
    df_hanmi            = fdr.DataReader(symbol = get_stock_code('한미반도체'), start = before_standard)[['Close']]

    df = pd.concat([df_exchange_rate, df_kospi, df_samsung_elec, df_samsung_pre_elec, df_sk_hynix, df_db_hitek, df_hanmi], axis = 1, join = 'inner')
    df.columns = ['USD/KRW', 'KOSPI', 'SS_ELEC', 'SS_ELEC_PRE', 'SK_HINIX', 'DB_HITEK', 'HANMI']
    df.reset_index(inplace = True)

    df_2_year = df.copy()
    df_1_year = df.copy()
    df_6_month = df.copy()
    df_3_month = df.copy()

    df_1_year = df_1_year[df_1_year['Date'] > str(datetime.now() - relativedelta(years = 1))]
    df_6_month = df_6_month[df_6_month['Date'] > str(datetime.now() - relativedelta(months = 6))]
    df_3_month = df_3_month[df_3_month['Date'] > str(datetime.now() - relativedelta(months = 3))]

    df_2_year['USD/KRW'] = round((df_2_year['USD/KRW'] - (df_2_year['USD/KRW'].sum() / df_2_year['Date'].count())) / df_2_year['USD/KRW'] * 100, 2)
    df_2_year['KOSPI'] = round((df_2_year['KOSPI'] - (df_2_year['KOSPI'].sum() / df_2_year['Date'].count())) / df_2_year['KOSPI'] * 100, 2)
    df_2_year['SS_ELEC'] = round((df_2_year['SS_ELEC'] - (df_2_year['SS_ELEC'].sum() / df_2_year['Date'].count())) / df_2_year['SS_ELEC'] * 100, 2)
    df_2_year['SS_ELEC_PRE'] = round((df_2_year['SS_ELEC_PRE'] - (df_2_year['SS_ELEC_PRE'].sum() / df_2_year['Date'].count())) / df_2_year['SS_ELEC_PRE'] * 100, 2)
    df_2_year['SK_HINIX'] = round((df_2_year['SK_HINIX'] - (df_2_year['SK_HINIX'].sum() / df_2_year['Date'].count())) / df_2_year['SK_HINIX'] * 100, 2)
    df_2_year['DB_HITEK'] = round((df_2_year['DB_HITEK'] - (df_2_year['DB_HITEK'].sum() / df_2_year['Date'].count())) / df_2_year['DB_HITEK'] * 100, 2)
    df_2_year['HANMI'] = round((df_2_year['HANMI'] - (df_2_year['HANMI'].sum() / df_2_year['Date'].count())) / df_2_year['HANMI'] * 100, 2)

    df_1_year['USD/KRW'] = round((df_1_year['USD/KRW'] - (df_1_year['USD/KRW'].sum() / df_1_year['Date'].count())) / df_1_year['USD/KRW'] * 100, 2)
    df_1_year['KOSPI'] = round((df_1_year['KOSPI'] - (df_1_year['KOSPI'].sum() / df_1_year['Date'].count())) / df_1_year['KOSPI'] * 100, 2)
    df_1_year['SS_ELEC'] = round((df_1_year['SS_ELEC'] - (df_1_year['SS_ELEC'].sum() / df_1_year['Date'].count())) / df_1_year['SS_ELEC'] * 100, 2)
    df_1_year['SS_ELEC_PRE'] = round((df_1_year['SS_ELEC_PRE'] - (df_1_year['SS_ELEC_PRE'].sum() / df_1_year['Date'].count())) / df_1_year['SS_ELEC_PRE'] * 100, 2)
    df_1_year['SK_HINIX'] = round((df_1_year['SK_HINIX'] - (df_1_year['SK_HINIX'].sum() / df_1_year['Date'].count())) / df_1_year['SK_HINIX'] * 100, 2)
    df_1_year['DB_HITEK'] = round((df_1_year['DB_HITEK'] - (df_1_year['DB_HITEK'].sum() / df_1_year['Date'].count())) / df_1_year['DB_HITEK'] * 100, 2)
    df_1_year['HANMI'] = round((df_1_year['HANMI'] - (df_1_year['HANMI'].sum() / df_1_year['Date'].count())) / df_1_year['HANMI'] * 100, 2)

    df_6_month['USD/KRW'] = round((df_6_month['USD/KRW'] - (df_6_month['USD/KRW'].sum() / df_6_month['Date'].count())) / df_6_month['USD/KRW'] * 100, 2)
    df_6_month['KOSPI'] = round((df_6_month['KOSPI'] - (df_6_month['KOSPI'].sum() / df_6_month['Date'].count())) / df_6_month['KOSPI'] * 100, 2)
    df_6_month['SS_ELEC'] = round((df_6_month['SS_ELEC'] - (df_6_month['SS_ELEC'].sum() / df_6_month['Date'].count())) / df_6_month['SS_ELEC'] * 100, 2)
    df_6_month['SS_ELEC_PRE'] = round((df_6_month['SS_ELEC_PRE'] - (df_6_month['SS_ELEC_PRE'].sum() / df_6_month['Date'].count())) / df_6_month['SS_ELEC_PRE'] * 100, 2)
    df_6_month['SK_HINIX'] = round((df_6_month['SK_HINIX'] - (df_6_month['SK_HINIX'].sum() / df_6_month['Date'].count())) / df_6_month['SK_HINIX'] * 100, 2)
    df_6_month['DB_HITEK'] = round((df_6_month['DB_HITEK'] - (df_6_month['DB_HITEK'].sum() / df_6_month['Date'].count())) / df_6_month['DB_HITEK'] * 100, 2)
    df_6_month['HANMI'] = round((df_6_month['HANMI'] - (df_6_month['HANMI'].sum() / df_6_month['Date'].count())) / df_6_month['HANMI'] * 100, 2)

    df_3_month['USD/KRW'] = round((df_3_month['USD/KRW'] - (df_3_month['USD/KRW'].sum() / df_3_month['Date'].count())) / df_3_month['USD/KRW'] * 100, 2)
    df_3_month['KOSPI'] = round((df_3_month['KOSPI'] - (df_3_month['KOSPI'].sum() / df_3_month['Date'].count())) / df_3_month['KOSPI'] * 100, 2)
    df_3_month['SS_ELEC'] = round((df_3_month['SS_ELEC'] - (df_3_month['SS_ELEC'].sum() / df_3_month['Date'].count())) / df_3_month['SS_ELEC'] * 100, 2)
    df_3_month['SS_ELEC_PRE'] = round((df_3_month['SS_ELEC_PRE'] - (df_3_month['SS_ELEC_PRE'].sum() / df_3_month['Date'].count())) / df_3_month['SS_ELEC_PRE'] * 100, 2)
    df_3_month['SK_HINIX'] = round((df_3_month['SK_HINIX'] - (df_3_month['SK_HINIX'].sum() / df_3_month['Date'].count())) / df_3_month['SK_HINIX'] * 100, 2)
    df_3_month['DB_HITEK'] = round((df_3_month['DB_HITEK'] - (df_3_month['DB_HITEK'].sum() / df_3_month['Date'].count())) / df_3_month['DB_HITEK'] * 100, 2)
    df_3_month['HANMI'] = round((df_3_month['HANMI'] - (df_3_month['HANMI'].sum() / df_3_month['Date'].count())) / df_3_month['HANMI'] * 100, 2)

    plt.figure(figsize=(20, 10))

    plt.subplot(2,2,1)
    plt.title('before 2 years')
    plt.plot(df_2_year['Date'], df_2_year['USD/KRW'].to_list(), label = 'USD/KRW')
    plt.plot(df_2_year['Date'], df_2_year['KOSPI'].to_list(), label = 'KOSPI', linestyle = '--')
    plt.plot(df_2_year['Date'], df_2_year['SS_ELEC'].to_list(), label = 'SS_ELEC')
    plt.plot(df_2_year['Date'], df_2_year['SS_ELEC_PRE'].to_list(), label = 'SS_ELEC_PRE')
    plt.plot(df_2_year['Date'], df_2_year['SK_HINIX'].to_list(), label = 'SK_HINIX')
    plt.plot(df_2_year['Date'], df_2_year['DB_HITEK'].to_list(), label = 'DB_HITEK')
    plt.plot(df_2_year['Date'], df_2_year['HANMI'].to_list(), label = 'HANMI')
    plt.grid(axis = 'x')
    plt.grid(axis = 'y')
    plt.legend()

    plt.subplot(2,2,2)
    plt.title('before 1 years')
    plt.plot(df_1_year['Date'], df_1_year['USD/KRW'].to_list(), label = 'USD/KRW')
    plt.plot(df_1_year['Date'], df_1_year['KOSPI'].to_list(), label = 'KOSPI', linestyle = '--')
    plt.plot(df_1_year['Date'], df_1_year['SS_ELEC'].to_list(), label = 'SS_ELEC')
    plt.plot(df_1_year['Date'], df_1_year['SS_ELEC_PRE'].to_list(), label = 'SS_ELEC_PRE')
    plt.plot(df_1_year['Date'], df_1_year['SK_HINIX'].to_list(), label = 'SK_HINIX')
    plt.plot(df_1_year['Date'], df_1_year['DB_HITEK'].to_list(), label = 'DB_HITEK')
    plt.plot(df_1_year['Date'], df_1_year['HANMI'].to_list(), label = 'HANMI')
    plt.grid(axis = 'x')
    plt.grid(axis = 'y')
    plt.legend()

    plt.subplot(2,2,3)
    plt.title('before 6 months')
    plt.plot(df_6_month['Date'], df_6_month['USD/KRW'].to_list(), label = 'USD/KRW')
    plt.plot(df_6_month['Date'], df_6_month['KOSPI'].to_list(), label = 'KOSPI', linestyle = '--')
    plt.plot(df_6_month['Date'], df_6_month['SS_ELEC'].to_list(), label = 'SS_ELEC')
    plt.plot(df_6_month['Date'], df_6_month['SS_ELEC_PRE'].to_list(), label = 'SS_ELEC_PRE')
    plt.plot(df_6_month['Date'], df_6_month['SK_HINIX'].to_list(), label = 'SK_HINIX')
    plt.plot(df_6_month['Date'], df_6_month['DB_HITEK'].to_list(), label = 'DB_HITEK')
    plt.plot(df_6_month['Date'], df_6_month['HANMI'].to_list(), label = 'HANMI')
    plt.grid(axis = 'x')
    plt.grid(axis = 'y')
    plt.legend()

    plt.subplot(2,2,4)
    plt.title('before 3 months')
    plt.plot(df_3_month['Date'], df_3_month['USD/KRW'].to_list(), label = 'USD/KRW')
    plt.plot(df_3_month['Date'], df_3_month['KOSPI'].to_list(), label = 'KOSPI', linestyle = '--')
    plt.plot(df_3_month['Date'], df_3_month['SS_ELEC'].to_list(), label = 'SS_ELEC')
    plt.plot(df_3_month['Date'], df_3_month['SS_ELEC_PRE'].to_list(), label = 'SS_ELEC_PRE')
    plt.plot(df_3_month['Date'], df_3_month['SK_HINIX'].to_list(), label = 'SK_HINIX')
    plt.plot(df_3_month['Date'], df_3_month['DB_HITEK'].to_list(), label = 'DB_HITEK')
    plt.plot(df_3_month['Date'], df_3_month['HANMI'].to_list(), label = 'HANMI')
    plt.grid(axis = 'x')
    plt.grid(axis = 'y')
    plt.legend()

    plt.show()