每天重复这样的Excel操作,写个python脚本多省事!

关注公众号:【数据分析与统计学之美】,可以进群和作者交流!

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()

结果如下:
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/weixin_41261833/article/details/106583693