Skyze/Graph2.py at develop · mikenew/Skyze · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
'''
Created on 06/09/2017

@author: michaelnew

From https://pythonprogramming.net/candlestick-ohlc-graph-matplotlib-tutorial/

'''


import pandas as pd
import numpy as np

# Possible fix
# import matplotlib
# matplotlib.use("gtk")

import matplotlib.pyplot as plt
import matplotlib.ticker as mticker
from matplotlib.finance import candlestick_ohlc
import matplotlib.dates as mdates

import pylab


import datetime as dt

from Skyze_Standard_Library.Market import Market

class GraphCandlestick2(object):
    '''
    classdocs
    '''


    def __init__(self):
        '''
        Constructor
        '''


    def graph_data(self,p_df_ohlc):
#         print("DF_OHLC .. before reset index"); print(p_df_ohlc.head(5))
        #Reset the index to remove Date column from index
        #df_ohlc = p_df_ohlc.reset_index()
        df_ohlc = p_df_ohlc
#         print("DF_OHLC .. after reset index"); print(df_ohlc.head(5))

        #Naming columns
        df_ohlc.columns = ["Date","Open","High",'Low',"Close"]

        #Converting dates column to float values
        df_ohlc['Date'] = df_ohlc['Date'].map(mdates.date2num)

        #Making plot
        fig = plt.figure()
        ax1 = plt.subplot2grid((6,1), (0,0), rowspan=6, colspan=1)

        #Converts raw mdate numbers to dates
        ax1.xaxis_date()
        plt.xlabel("Date")
        print("DF_OHLC .. after date2num in graph_data"); print(df_ohlc.head(5))

        #Making candlestick plot
        candlestick_ohlc(ax1,df_ohlc.values,width=1, colorup='g', colordown='k',alpha=0.75)
        # Parameters:
#             ax : Axes... an Axes instance to plot to
#             quotes :    sequence of (time, open, high, low, close, ...) sequences
#                         As long as the first 5 elements are these values, the record can be as long as you want (e.g., it may store volume).
#                         time must be in float days format - see date2num
#             width : float...fraction of a day for the rectangle width
#             colorup : color ... the color of the rectangle where close >= open
#             colordown : color ...the color of the rectangle where close < open
#             alpha : float ... the rectangle alpha level
        plt.ylabel("Price")
        plt.legend()

#         fig.autofmt_xdate() #??
        plt.close('all')
        print("Mike - about to plt.show")
        plt.show()

    def testGraph(self):
        data = [1,1,2,3,5,8,13,21,34]

#         plt.figure
#         plt.title('mike', size = 'xx-large')
#         plt.plot(data)
#         plt.show()
#         %matplotlib inline
        plt.ion()
        plt.plot(data)
        pylab.show()


if __name__ == "__main__":

    mkt_name = "Bitcoin_Test"
    mkt = Market.fromTesting(mkt_name)
    mkt_data = mkt.readMarketDataCSV(p_testing=True)

    graph = GraphCandlestick2()
    quotes = mkt_data.iloc[0:3,:4]
    print("quotes"); print(quotes.head(5))
    #graph.graph_data(quotes)
    graph.testGraph()

    plt.switch_backend("macosx")
    plt.plot([1,2,3,4])
    plt.ylabel('some numbers')
    plt.show()
    print("plotted")

#     Backend    Description
# GTKAgg    Agg rendering to a GTK 2.x canvas (requires PyGTK and pycairo or cairocffi; Python2 only)
# GTK3Agg    Agg rendering to a GTK 3.x canvas (requires PyGObject and pycairo or cairocffi)
# GTK    GDK rendering to a GTK 2.x canvas (not recommended and d eprecated in 2.0) (requires PyGTK and pycairo or cairocffi; Python2 only)
# GTKCairo    Cairo rendering to a GTK 2.x canvas (requires PyGTK and pycairo or cairocffi; Python2 only)
# GTK3Cairo    Cairo rendering to a GTK 3.x canvas (requires PyGObject and pycairo or cairocffi)
# WXAgg    Agg rendering to to a wxWidgets canvas (requires wxPython)
# WX    Native wxWidgets drawing to a wxWidgets Canvas (not recommended and deprecated in 2.0) (requires wxPython)
# TkAgg    Agg rendering to a Tk canvas (requires TkInter)
# Qt4Agg    Agg rendering to a Qt4 canvas (requires PyQt4 or pyside)
# Qt5Agg    Agg rendering in a Qt5 canvas (requires PyQt5)
# macosx    Cocoa rendering in OSX windows (presently lacks blocking show() behavior when matplotlib is in non-interactive mode)


# matplotlib.finance.candlestick_ohlc(ax, quotes, width=0.2, colorup='k', colordown='r', alpha=1.0)
#
# Parameters:
#             ax : Axes... an Axes instance to plot to
#             quotes :    sequence of (time, open, high, low, close, ...) sequences
#                         As long as the first 5 elements are these values, the record can be as long as you want (e.g., it may store volume).
#                         time must be in float days format - see date2num
#             width : float...fraction of a day for the rectangle width
#             colorup : color ... the color of the rectangle where close >= open
#             colordown : color ...the color of the rectangle where close < open
#             alpha : float ... the rectangle alpha level
#
# Returns:    ret : tuple
#                 returns (lines, patches) where lines is a list of lines added and patches is a list of the rectangle patches added