1. 提出需求
这是群里面一位朋友提出的一个需求,具体需求是啥样的?有这样一个工作簿,我们要针对sheet1表中的数据,筛选出【地区==深证】的数据,将其写入sheet2表中。与此同时,sheet1的数据就变成了【非深圳】的那些数据。
2. 原始数据预览
观察下面的原始数据可以发现:这个工作簿目前只有一个sheet1表,表中有一些样本数据。我们此时要做的,就是筛选【地区==深证】的数据,将其写入同一个工作簿的sheet2表中。同时这个sheet1表中的数据,最后要变成【去掉深证后】的数据。
3. 解决问题
我的思路大致是这样的:因为该需求对于数据的操作,都必须是在这同一个工作簿中进行。因此我先将“筛选出来的深圳数据”写入到sheet2中,然后删除原始的sheet1表,最后再将“去掉深证后的数据”写入到sheet1中,大功告成!
import pandas as pd from openpyxl import load_workbook df = pd.read_excel(r"G:\1微信公众号图片素材\写好的文章\工作簿 (1).xlsx",sheet_name='Sheet1') # 筛选得到“非深圳”的数据 df1 = df[df["地区"] != "深圳"] # 筛选得到“都是深圳”的数据 df2 = df[df["地区"] == "深圳"] # 在ExcelWriter的源代码中,它会初始化得到一个空的工作簿 writer = pd.ExcelWriter(r"G:\1微信公众号图片素材\写好的文章\工作簿 (1).xlsx",engine='openpyxl') # print(writer.path) # 利用openpyxl读取工作簿 (1).xlsx工作簿 workbook = load_workbook(writer.path) # 动态添加一个工作簿,此时writer就不是一个空的工作簿了 writer.book = workbook # 将“都是深圳”的数据,首先写入到“sheet2表”中 df2.to_excel(excel_writer=writer,sheet_name="Sheet2",index=False) # 删除原始的“sheet1表” sheet = workbook["Sheet1"] workbook.remove(sheet) workbook.save(filename = r"G:\1微信公众号图片素材\写好的文章\工作簿 (1).xlsx") # 再将“非深圳”的数据,写入到“sheet1表”中 df1.to_excel(excel_writer=writer,sheet_name="Sheet1",index=False) writer.save() writer.close()
结果如下: