pandas write to Excel with chart

##############################################################################

#

# An example of converting a Pandas dataframe to an xlsx file with a chart

# using Pandas and XlsxWriter.

#

import pandas as pd

# Create a Pandas dataframe from some data.

df = pd.DataFrame({'Data': [10, 20, 30, 20, 15, 30, 45]})

# Create a Pandas Excel writer using XlsxWriter as the engine.

writer = pd.ExcelWriter('pandas_chart.xlsx', engine='xlsxwriter')

# Convert the dataframe to an XlsxWriter Excel object.

df.to_excel(writer, sheet_name='Sheet1')

# Get the xlsxwriter workbook and worksheet objects.

workbook = writer.book

worksheet = writer.sheets['Sheet1']

# Apply a conditional format to the cell range.

worksheet.conditional_format('B2:B8', {'type': '3_color_scale'})

# Create a chart object.

chart = workbook.add_chart({'type': 'column'})

# Configure the series of the chart from the dataframe data.

chart.add_series({'values': '=Sheet1!$B$2:$B$8'})

# Insert the chart into the worksheet.

worksheet.insert_chart('D2', chart)

# Close the Pandas Excel writer and output the Excel file.

writer.save()