将新工作表添加到现有 Excel 文件并绘制数据图表答案

【问题标题】：Add a new sheet to existing excel file and graph the data将新工作表添加到现有 Excel 文件并绘制数据图表
【发布时间】：2019-07-31 05:26:23
【问题描述】：

我希望通过导出 pandas 数据框并使用 openpyxl 绘制数据来将新工作表添加到 excel 文件中。目前我可以添加一个新工作表并插入熊猫数据框，但不能插入数据图。

目前我的代码如下所示。

import pandas
import openpyxl
from openpyxl.chart import LineChart, Reference

fileSavePath = 'C:/Users/PATH'
filename = "Existing Excel File"

lst1 = [1, 2, 3, 4, 5, 6, 7, 8, 9]
lst2 = [0, 5, 2, 8, 15, 7, 10, 4, 0]

df = pandas.DataFrame({'First Column': lst1, 'Second Column': lst2})

book = openpyxl.load_workbook(fileSavePath + filename + '.xlsx')
with pandas.ExcelWriter(fileSavePath + filename + '.xlsx', engine='openpyxl') as writer:
    writer.book = book
    writer.sheets = dict((ws.title, ws) for ws in book.worksheets)
    df.to_excel(writer, 'new_sheet', index=False)

wb = openpyxl.Workbook()
sheet = wb.active
values = Reference(sheet, min_col=1, min_row=1,
                   max_col=1, max_row=len(lst1))
chart = LineChart()
chart.add_data(values)
chart.title = " LINE-CHART "
chart.x_axis.title = " X-AXIS "
chart.y_axis.title = " Y-AXIS "
sheet.add_chart(chart, "E2")
writer.save()
writer.close()

目前没有错误，数据框写入但图表没有出现在新添加的工作表中。

【问题讨论】：

标签： python pandas openpyxl

【解决方案1】：

此代码将成功添加新工作表并绘制数据图表。

import pandas
from openpyxl.chart import LineChart, Reference
from openpyxl.utils.dataframe import dataframe_to_rows
from openpyxl import load_workbook

lst1 = [1, 2, 3, 4, 5, 6, 7, 8, 9]
lst2 = [0, 5, 2, 8, 15, 7, 10, 4, 0]
df = pandas.DataFrame({'First Column': lst1, 'Second Column': lst2})
fileSavePath = 'C:/Users/PathExample'
filename = "filename"
newSheetName = 'newSheet'
sheet = newSheetName
# load in existing workbook
wb = load_workbook(fileSavePath + filename + '.xlsx')
# create new sheet with name assigned by title
wb.create_sheet(newSheetName)
# find which sheet is the new sheet
newSheetIndex = len(wb.sheetnames) - 1
# set the newest sheet as the active sheet
wb.active = newSheetIndex
worksheet = wb.active

# Insert Dataframe
for r in dataframe_to_rows(df, index=False, header=True):
    worksheet.append(r)

#Graph data
chart = LineChart()
data = Reference(worksheet, range_string=f'{sheet}!B2:B10')
category = Reference(worksheet, range_string=f'{sheet}!A2:A10')
chart.add_data(data, titles_from_data=True)
chart.set_categories(category)
worksheet.add_chart(chart, 'D3')
wb.save(fileSavePath + filename + '.xlsx')

问题似乎在于设置活动工作表。前面的代码添加了工作表，但一直在绘制“活动工作表”，这就是该图表没有出现在其他工作表上的原因。 Openpyxl 使用一个列表来跟踪 excel 工作表，因此要操作所需的工作表，您需要将该索引号设置为活动工作表 (wb.active)。

【讨论】：