根据excel中任意一列的列名进行拆分--Python

需求

在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进行优化。大家可以将自己需要的需求写在下面，有时间可以写一写，提高自己的工作效率。

关注我的公众号