[Py2.7] Candlestick OHLC graphs(촛대 그래프) w/ Matplotlib.pyplot

/*******************************************************************************************************************
-- Title : [Py2.7] Candlestick OHLC graphs(촛대 그래프) w/ Matplotlib.pyplot
-- Reference : pythonprogramming.net
-- Key word : python 파이썬 matplotlib pyplot finance candlestick ohlc numpy ticker dates urllib datetime
                  촛대 차트 촛대 그래프
*******************************************************************************************************************/

-- Chart

-- Python

# -*- coding: utf-8 -*-
 
import matplotlib.pyplot as plt
import matplotlib.dates as mdates
import matplotlib.ticker as mticker
from matplotlib.finance import candlestick_ohlc
 
import numpy as np
import urllib
import datetime as dt
 
 
def bytespdate2num(fmt, encoding='utf-8'):
    strconverter = mdates.strpdate2num(fmt)
 
    def bytesconverter(b):
        s = b.decode(encoding)
        return strconverter(s)
 
    return bytesconverter
 
def graph_data(stock):
    fig = plt.figure()
    ax1 = plt.subplot2grid((1, 1), (0, 0))
 
    stock_price_url = 'http://chartapi.finance.yahoo.com/instrument/1.0/' + stock + '/chartdata;type=quote;range=1m/csv'
    #print stock_price_url
 
    # -- URL 통해 데이터소스 가져오기
    source_code = urllib.urlopen(stock_price_url).read()
    #print source_code
 
    # -- '\n' 기준으로 분리
    stock_data = []
    split_source = source_code.split('\n')
    #print split_source
 
    for line in split_source:
        split_line = line.split(',')                                      # 콤마(,)를 기준으로 분리
        #print split_line
 
        if len(split_line) == 6:
            if 'values' not in line and 'labels' not in line:             # labels:values 형태이면 SKIP
                stock_data.append(line)
    #print stock_data
    #print "-" * 100
 
    # -- 리스트에 있는 항목을 np.loadtxt로 구분하여 저장
    date, closep, highp, lowp, openp, volume = np.loadtxt(stock_data,
                                                          delimiter=',',
                                                          unpack=True,
                                                          converters={0: bytespdate2num('%Y%m%d')})
 
    print type(date)
 
    x = 0
    y = len(date)
    ohlc = []
 
    # -- numpy를 tuple로 저장
    while x < y:
        append_me = date[x], openp[x], highp[x], lowp[x], closep[x], volume[x]
 
        ohlc.append(append_me)
        x += 1
 
    print type(append_me)
 
    candlestick_ohlc(ax1, ohlc, width=0.4, colorup='#77d879', colordown='#db3f3f')
 
    for label in ax1.xaxis.get_ticklabels():
        label.set_rotation(45)
 
    ax1.xaxis.set_major_formatter(mdates.DateFormatter('%Y-%m-%d'))
    ax1.xaxis.set_major_locator(mticker.MaxNLocator(10))
    ax1.grid(True)
 
    plt.xlabel('xlabel:Date')
    plt.ylabel('ylabel:Price')
    plt.title(stock)
    plt.legend()                                                          # warning이유 모르겠다!
    plt.subplots_adjust(left=0.09, bottom=0.20, right=0.94, top=0.90, wspace=0.2, hspace=0)
    plt.show()
 
 
graph_data('EBAY')