第5节,pandas实现execl筛选数据

在这张excel表中存储了员工信息,需要你根据性别筛选数据,将所有男员工存储在一个sheet中,所有女员工存储在一个sheet中;再以年龄划分,将大于等于20岁的员工存储在一个sheet中,小于20岁的员工存储在一个sheet中。

第一步,打开excel

import pandas as pd

result_file = "./data/数据筛选.xlsx"
df = pd.read_excel(result_file)

第二步,创建以openpyxl为引擎的ExcelWriter对象

target_excel = "./data/员工筛选.xlsx"
writer = pd.ExcelWriter(target_excel, engine='openpyxl')

第三步,筛选数据并写入excel

man_df = df[df['性别'] == "男"]
woman_df = df[df['性别'] == "女"]
ge_20_df = df[df['年龄'] >= 20]
lt_20_df = df[df['年龄'] < 20]

man_df.to_excel(writer, sheet_name="男员工", index=False)
woman_df.to_excel(writer, sheet_name="女员工", index=False)
ge_20_df.to_excel(writer, sheet_name="年龄大于等于20", index=False)
lt_20_df.to_excel(writer, sheet_name="年龄小于20", index=False)
writer.save()

扫描关注, 与我技术互动

QQ交流群: 211426309

加入知识星球, 每天收获更多精彩内容

分享日常研究的python技术和遇到的问题及解决方案