需求
在excel中,根据“状态”进行分割,两种解决方法,一种是拆分保存在各个Sheet表中,另外一种拆分保存在各个文件中。
具体代码如下
需要导入的包
import pandas as pd
import os
1. 拆分保存在各个Sheet表
# 根据指定列信息分割多个sheet
def spilt_into_multiple_sheets(df, spilt_column, output_file):
with pd.ExcelWriter(output_file) as writer:
for group_name, group_info in df.groupby([spilt_column]):
group_info.to_excel(writer, sheet_name="" + str(group_name), index=False)
print("成功")
2. 拆分保存在各个文件中
# 根据指定列信息分割多个文件
def spilt_into_multiple_files(df, spilt_column, output_folder):
if not os.path.exists(output_folder):
os.makedirs(output_folder)
for group_name, group_info in df.groupby([spilt_column]):
group_info.to_excel(output_folder + "/" + group_name + ".xlsx", sheet_name='sheet1', index=False)
print("成功")
3. 调用上面函数
import pandas as pd
import os
df = pd.read_excel("bugsOutput.xlsx")
# df, column name, output file name
spilt_into_multiple_sheets(df, "状态", "output_file.xlsx")
# df, column name, output folder name
spilt_into_multiple_files(df, "状态", "output_folder")
结果
各个Sheet表
各个文件中
More
在实际工作中,我们有很多重复性的工作,这些工作可以试图用Python进行优化。大家可以将自己需要的需求写在下面,有时间可以写一写,提高自己的工作效率。
关注我的公众号