Python | Plotting Stock charts in excel sheet using XlsxWriter module

Prerequisite: Create and Write on an excel sheet

XlsxWriter is a Python library using which one can perform multiple operations on excel files like creating, writing, arithmetic operations and plotting graphs. Let’s see how to plot Stock charts using realtime data.

Charts are composed of at least one series of one or more data points. Series themselves are comprised of references to cell ranges. For plotting the charts on an excel sheet, firstly, create chart object of specific chart type( i.e Stock chart etc.). After creating chart objects, insert data in it and lastly, add that chart object in the sheet object.

Code : Plot the Stock Chart.
For plotting the Stock chart on an excel sheet, use add_chart() method with type ‘stock’ keyword argument of a workbook object.

filter_none

edit
close

play_arrow

link
brightness_4
code

# 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.
# Add a data series to a chart 
# 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()

chevron_right


 

Output:
output - 1



My Personal Notes arrow_drop_up

Check out this Author's contributed articles.

If you like GeeksforGeeks and would like to contribute, you can also write an article using contribute.geeksforgeeks.org or mail your article to contribute@geeksforgeeks.org. See your article appearing on the GeeksforGeeks main page and help other Geeks.

Please Improve this article if you find anything incorrect by clicking on the "Improve Article" button below.