12-openpyxl 追加 Excel 数据
1. 需求说明
你好，我是悦创。
xlsxwriter 只能创建新的文件，不能对文件进行数据的追加和读取操作。
需要追加和读取 xlsx 文件，就需要用到 openpyxl 这个库了。安装命令：pip install openpyxl。
pip install openpyxl使用 openpyxl 完成这面这个需求：
- 读取上面创建好的
xlsxwriter插入数据和折线图.xlsx文件，复制一份，保存到 poenpyxl 插入数据和折线图[copy xlsxwriter].xlsx复制的文件中 - sheet1 保持和源文件的 sheet1 一致，折线图不画复制的文件中
- 创建一个 sheet2，sheet2 的数据从 sheet1 中拷贝过来
- 随机的增加 1 年的随机数据，也就是 2020 年 1 月至 12 月的日期，数据1和数据2的12个数据随机生成
- 在 sheet2 中画一个折线图，统计数据1和数据2
需求有四个，有难有易。
2. 拷贝文件
2.1 第一个是拷贝文件和 sheet1
其实拷贝了文件，自然就包含 sheet1，最简单了，如下代码：
import openpyxl
filename = 'xlsxwriter插入数据和折线图.xlsx'
wb = openpyxl.load_workbook(filename)
wb.save('poenpyxl插入数据和折线图[copy xlsxwriter].xlsx')这样就得到了源文件和 sheet1 的复制品。
3. 拷贝 sheet 内容
3.1 第二个需求，拷贝 sheet1，做成 sheet2
也简单，如下代码：
import openpyxl
filename = 'xlsxwriter插入数据和折线图.xlsx'
wb = openpyxl.load_workbook(filename)
sheet1 = wb['sheet1']
sheet2 = wb.copy_worksheet(sheet1)
sheet2.title = "sheet2"
wb.save('poenpyxl插入数据和折线图[copy xlsxwriter].xlsx')获取 sheet1，复制一个给 sheet2 参数，然后改下 sheet2 的标题 title，这个 title 就是文件的 sheet2 的名字。
4. 追加数据内容
4.1 第三个需求，sheet2 中，数据1 和数据2 追加一年的数据，数据可以随意生成
貌似简单，但是年份时间不好计算，并且现在只有 2019 年的。如果之前数据是 2018 年，则追加一年，应该是 2019 年，所以要读取时间和月份追加。
这里的难点就是时间字符串的读取，以及字符串转时间，以及时间的相加。
如下实现代码：
import openpyxl, random, datetime
from dateutil.relativedelta import relativedelta
filename = 'xlsxwriter插入数据和折线图.xlsx'
wb = openpyxl.load_workbook(filename)
sheet1 = wb['sheet1']
sheet2 = wb.copy_worksheet(sheet1)
sheet2.title = "sheet2"
rows = sheet2.max_row # 读取最后一行
prev_date_str = sheet2.cell(row=rows,column=1).value # 取出时间的字符串
prev_date = datetime.datetime.strptime(prev_date_str, "%Y-%m") # 时间字符串转时间对象
for i in range(1,13):
tmp_date = prev_date + relativedelta(months=i) # 月份的计算，每次增加一个月，就得到了第二年的12个月
tmp_num1 = random.randint(1,100)
tmp_num2 = random.randint(1,100)
sheet2.append([tmp_date.strftime("%Y-%m"), tmp_num1, tmp_num2])
wb.save('poenpyxl插入数据和折线图[copy xlsxwriter].xlsx')实现思路：
- 读取出最后一行
- 取出时间字符串，然后转换成时间对象
- 一年 12 个月，循环 12 次，每次增加一个月，数值可以随机的生成，用 random 即可
- 数据的追加，是按每行，所以将【时间， 数据1， 数据2】通过 append 直接追加到数据最后即可
第三个需求搞定，下一个需求，sheet2 画折线图
使用 openpyxl 画图表
第四个需求，在 sheet2 中对全部数据画折线图
这个难度不大，只要知道 openpyxl 的画图工具即可，如下代码：
import openpyxl, random, datetime
from dateutil.relativedelta import relativedelta
filename = 'xlsxwriter插入数据和折线图.xlsx'
wb = openpyxl.load_workbook(filename)
sheet1 = wb['sheet1']
sheet2 = wb.copy_worksheet(sheet1)
sheet2.title = "sheet2"
rows = sheet2.max_row
prev_date_str = sheet2.cell(row=rows,column=1).value
prev_date = datetime.datetime.strptime(prev_date_str, "%Y-%m")
for i in range(1,13):
tmp_date = prev_date + relativedelta(months=i)
tmp_num1 = random.randint(1,100)
tmp_num2 = random.randint(1,100)
sheet2.append([tmp_date.strftime("%Y-%m"), tmp_num1, tmp_num2])
# 下面是画折线图的实现代码
from openpyxl.chart import Series,LineChart, Reference
chart = LineChart() #图表对象
rows = sheet2.max_row
data1 = Reference(sheet2, min_col=2, min_row=1, max_col=2, max_row=rows) #涉及数据
title1 = sheet2.cell(row=1,column=2).value
seriesObj1 = Series(data1, title=title1) #创建series对象
data2 = Reference(sheet2, min_col=3, min_row=1, max_col=3, max_row=rows) #涉及数据
title2 = sheet2.cell(row=1,column=3).value
seriesObj2 = Series(data2, title=title2) #创建series对象
chart.append(seriesObj1) #添加到chart中
chart.append(seriesObj2) #添加到chart中
sheet2.add_chart(chart, "E3") #将图表添加到 sheet中
wb.save('poenpyxl插入数据和折线图[copy xlsxwriter].xlsx')导入所需的画图工具，图表初始化，然后生成数据对象：
- data1 的生成，因为索引从1开始，所以标题是第一行第二列，数据是第二行第二列，一直到最后一行第二列
- data2 的生成，因为索引从1开始，所以标题是第一行第三列，数据是第二行第三列，一直到第二行最后三列
- 将两个数据都放到图表内
- 然后图表的开始位置，设置成 E3，数据在 ABC，E 是空的，3 距离顶部有两格的位置
最后文件保存，大功告成。
查看最后的效果图
然后就是用 Office 打开 poenpyxl 插入数据和折线图[copy xlsxwriter].xlsx，看下 sheet2 的样子，如下图：
25 行数据，除了标题 24 行，刚好是 2019+2020 的 24 个月。折线图也一切正常。
欢迎关注我公众号：AI悦创，有更多更好玩的等你发现！
公众号：AI悦创【二维码】
AI悦创·编程一对一
AI悦创·推出辅导班啦，包括「Python 语言辅导班、C++ 辅导班、java 辅导班、算法/数据结构辅导班、少儿编程、pygame 游戏开发」，全部都是一对一教学：一对一辅导 + 一对一答疑 + 布置作业 + 项目实践等。当然，还有线下线上摄影课程、Photoshop、Premiere 一对一教学、QQ、微信在线，随时响应！微信：Jiabcdefh
C++ 信息奥赛题解，长期更新！长期招收一对一中小学信息奥赛集训，莆田、厦门地区有机会线下上门，其他地区线上。微信：Jiabcdefh
方法一：QQ
方法二：微信：Jiabcdefh
