Last active
June 10, 2019 08:21
-
-
Save benyblack/294fad56a2cad2207a7f77027175c658 to your computer and use it in GitHub Desktop.
Stock chart with oscillator using bokeh
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import numpy as np | |
from bokeh.layouts import column | |
from bokeh.plotting import figure, show, output_file | |
# imports for the example | |
import random | |
from bokeh.sampledata.stocks import AAPL, GOOG, IBM, MSFT | |
def datetime(x): | |
return np.array(x, dtype=np.datetime64) | |
def create_plot(chart_title): | |
plot = figure(x_axis_type="datetime", title=chart_title, plot_height=600, width=1000) | |
plot.grid.grid_line_alpha = 0.3 | |
plot.xaxis.axis_label = 'Date' | |
plot.yaxis.axis_label = 'Price' | |
return plot | |
def create_plot_oscillator(related_plot): | |
plot = figure(x_axis_type="datetime", x_range=related_plot.x_range, plot_height=150, width=1000) | |
plot.grid.grid_line_alpha = 0.3 | |
plot.xaxis.axis_label = 'Date' | |
plot.yaxis.axis_label = 'Price' | |
return plot | |
def add_line_to_plot(plot, data_title, data_date, data_close, color): | |
plot.line(data_date, data_close, color=color, legend=data_title) | |
return plot | |
def add_up_to_plot(plot, data_title, data_date, data_close, color="green"): | |
plot.triangle(x=data_date, y=data_close, color=color, legend=data_title) | |
return plot | |
def add_down_to_plot(plot, data_title, data_date, data_close, color="red"): | |
plot.inverted_triangle(x=data_date, y=data_close, | |
color=color, legend=data_title) | |
return plot | |
def show_plot(plot, oscillator=None): | |
plot.legend.location = "top_left" | |
output_file("stocks.html", title="stocks.py example") | |
if oscillator is None: | |
show(column(children=[plot])) | |
else: | |
show(column(children=[plot, oscillator])) | |
# Example of how to use this code | |
plot = create_plot("My chart title") | |
new_plot = add_line_to_plot(plot, "APPL", datetime( | |
AAPL['date']), AAPL['adj_close'], '#33A02C') | |
new_plot = add_up_to_plot(plot, "GOOG", datetime(GOOG['date']), GOOG['adj_close']) | |
new_plot = add_down_to_plot(plot, "MSFT", datetime(MSFT['date']), MSFT['adj_close']) | |
oscillator = create_plot_oscillator(new_plot) | |
indicator = [] | |
for x in IBM['adj_close']: | |
indicator.append(random.random()*100) | |
add_line_to_plot(oscillator, "IBM", datetime(IBM['date']), indicator, 'blue') | |
show_plot(new_plot, oscillator) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment