GeezerVison is showing github.com — SimpleStockAnalysisPython/Python_04_Plot_Charts.py at master · PCWin12/SimpleStockAnalysisPython

Expand file tree

Python_04_Plot_Charts.py

Copy path

More file actions

Latest commit

History

147 lines (117 loc) · 3.44 KB

Python_04_Plot_Charts.py

Copy path

File metadata and controls

147 lines (117 loc) · 3.44 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

# -*- coding: utf-8 -*-

"""

Created on Fri Nov 27 08:09:11 2020

@author: Tin

"""

# Plot Line Charts

import pandas as pd # Dataframe Library

import matplotlib.pyplot as plt # Plot Chart Library

pd.set_option('max_columns', None) # To show all columns

import yfinance as yf

yf.pdr_override()

# input

symbol = 'AAPL'

start = '2014-01-01'

end = '2018-01-01'

# dataframe

data = yf.download(symbol,start,end)

# View the first 5 rows

print('First 5 Rows')

print(data.head())

print('-'*80)

print('Line Chart')

plt.figure(figsize=(12,8))

plt.plot(data['Adj Close'])

plt.title("Stock Line Chart")

plt.legend(loc='best')

plt.xlabel("Date")

plt.ylabel("Price")

plt.show()

print('-'*80)

print('Line Chart with Grid')

plt.figure(figsize=(12,8))

plt.plot(data['Adj Close'])

plt.title("Stock Line Chart")

plt.grid()

plt.legend(loc='best')

plt.xlabel("Date")

plt.ylabel("Price")

plt.show()

print('-'*80)

print('Render the grid')

fig, ax = plt.subplots()

data.plot(kind='line', y= 'Adj Close', ax=ax)

# Turn on the grid

ax.grid()

plt.title("Stock Line Chart")

plt.legend(loc='best')

plt.xlabel("Date")

plt.ylabel("Price")

plt.show()

print('-'*80)

print('Customize the grid')

fig, ax = plt.subplots()

data.plot(kind='line', y= 'Adj Close', ax=ax)

# Don't allow the axis to be on top of your data

ax.set_axisbelow(True)

# Customize the grid

ax.grid(linestyle='-', linewidth='0.5', color='red')

plt.title("Stock Line Chart")

plt.xlabel("Date")

plt.ylabel("Price")

plt.show()

print('-'*80)

print('Major grid & Minor Grid')

plt.figure(figsize=(12,8))

plt.plot(data['Adj Close'])

plt.minorticks_on()

plt.grid(b=True, which='major', color='b', linestyle='-')

plt.grid(b=True, which='minor', color='r', linestyle='--')

plt.title("Stock Line Chart")

plt.xlabel("Date")

plt.ylabel("Price")

plt.show()

print('-'*80)

import seaborn as sns # Plot Library 0.9.0 Version

# conda install -c anaconda seaborn=0.9.0

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

sns.lineplot(data=data, x=data.index, y='Adj Close')

print('-'*80)

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

top = plt.subplot2grid((4,4), (0, 0), rowspan=3, colspan=4)

bottom = plt.subplot2grid((4,4), (3,0), rowspan=1, colspan=4)

top.plot(data.index, data['Adj Close'])

bottom.bar(data.index, data['Volume'])

# set the labels

top.axes.get_xaxis().set_visible(False)

top.set_title('Stock Price and Volume')

top.set_ylabel('Adj Closing Price')

bottom.set_ylabel('Volume')

print('-'*80)

# Candlestick

from mpl_finance import candlestick_ohlc

from matplotlib import dates as mdates

# Converting date to pandas datetime format

dfc = data.copy()

dfc = dfc.reset_index()

dfc['Date'] = pd.to_datetime(dfc['Date'])

dfc['Date'] = dfc['Date'].apply(mdates.date2num)

# dfc.head()

fig = plt.figure(figsize=(14,10))

ax1 = plt.subplot(2, 1, 1)

candlestick_ohlc(ax1,dfc.values, width=0.5, colorup='g', colordown='r', alpha=1.0)

ax1.xaxis_date()

ax1.xaxis.set_major_formatter(mdates.DateFormatter('%d-%m-%Y'))

ax1.set_title('Stock '+ symbol +' Closing Price')

ax1.set_ylabel('Price')

import plotly.graph_objs as go

# from plotly.offline import init_notebook_mode, iplot

df = data.copy()

# Plot OHLC Bar Chart

trace = go.Ohlc(x=df['12-2016'].index,

open=df['12-2016'].Open,

high=df['12-2016'].High,

low=df['12-2016'].Low,

close=df['12-2016'].Close)

data = [trace]

iplot(data, filename='simple_ohlc')