Python - Plotting Stock charts in excel sheet using XlsxWriter module

PythonServer Side ProgrammingProgramming

Stock chart is simply a price chart that shows a stock’s price plotted over a time frame, and it shows a few key sets of information −

  • Stock symbol and exchange.
  • Chart period
  • Price Change
  • Last Change
  • Volume

Example

# from datetime libraray import datetime method
from datetime import datetime
# import xlsxwriter library
import xlsxwriter
# Workbook() takes one, non-optional, argument which is the filename #that we want to create.
workbook = xlsxwriter.Workbook('chart_stock.xlsx')
# The workbook object is then used to add new worksheet via the #add_worksheet() method.  
worksheet = workbook.add_worksheet()
# Create a new Format object to formats cells in worksheets using #add_format() method .
# here we create bold format object .
bold = workbook.add_format({'bold': 1})
# here we create date format object
date_format = workbook.add_format({'num_format': 'dd/mm/yyyy'})
# Create a chart object that can be added to a worksheet using #add_chart() method.
# here we create a stock chart object .
chart = workbook.add_chart({'type': 'stock'})
# Add the worksheet data that the charts will refer to.
headings = ['Date', 'High', 'Low', 'Close']
data = [
   ['2018-01-01', '2018-01-02', '2018-01-03', '2018-01-04', '2018-01-05'],
   [27.2, 25.03, 19.05, 20.34, 18.5],
   [23.49, 19.55, 15.12, 17.84, 16.34],
   [25.45, 23.05, 17.32, 20.45, 17.34],
]
# Write a row of data starting from 'A1' with bold format .
worksheet.write_row('A1', headings, bold)
# Filling the data in an excel sheet
for row in range(5):
   # convert a string type date into date of rspective format.
   date = datetime.strptime(data[0][row], "% Y-% m-% d")
   # apply a date_format object on this particular cell
   worksheet.write(row + 1, 0, date, date_format)
   # write data in the respective cells
   worksheet.write(row + 1, 1, data[1][row])
   worksheet.write(row + 1, 2, data[2][row])
   worksheet.write(row + 1, 3, data[3][row])
# width of A to D column set to 11.
worksheet.set_column('A:D', 11)
# Add a series for each of the High-Low-Close columns.
# using add_series method.
# note : spaces is not inserted in b / w
# = and Sheet1, Sheet1 and !
# if space is inserted it throws warning.
# data take from A2 to A6 and B2 to B6 respectively.
chart.add_series({
   'categories': '= Sheet1 !$A$2:$A$6',
   'values': '= Sheet1 !$B$2:$B$6',
})  
# data take from A2 to A6 and C2 to C6 respectively.
chart.add_series({
   'categories': '= Sheet1 !$A$2:$A$6',
   'values':     '= Sheet1 !$C$2:$C$6',
})  
# data take from A2 to A6 and D2 to D6 respectively.
chart.add_series({
   'categories': '= Sheet1 !$A$2:$A$6',
   'values': '= Sheet1 !$D$2:$D$6',
})  
# Add a chart title  
chart.set_title ({'name': 'High-Low-Close'})  
# Add x-axis label
chart.set_x_axis({'name': 'Date'})  
# Add y-axis label
chart.set_y_axis({'name': 'Share price'})  
# add chart to the worksheet with given offset values at the top-#left corner of a chart is anchored to cell E9
worksheet.insert_chart('E9', chart)  
# Finally, close the Excel file via the close() method.  
workbook.close()
raja
Published on 06-Aug-2020 16:41:46
Advertisements