【Python】处理Excel中数据2 (按照班级筛选数据_生成多个文件)

一，源数据如下

需求：1. 按照班级把同一个班级的所有成员筛选出来。

　　　2. 筛选出来的数据作为单独的Excel表生成保存

二，代码的编写

# -*- coding:utf-8 -*-

import openpyxl

# 1.加载Excel源数据
path = R"C:UsersAdministratorpython处理Excel数据20200113按照班级分离文件.xlsx"
workbook = openpyxl.load_workbook(path)
sheet_names = workbook.sheetnames
sheet1 = workbook[sheet_names[0]]

# 2. 读取Excel sheet1中的所有数据
allDatas = []
for row in sheet1.rows:
    lines = [cell.value for cell in row]
    allDatas.append(lines)

#划分班级
title = []
class1 = []
class2 = []
class3 = []
class4 = []
class5 = []

for i in allDatas:
    if i[0] == "1":
        class1.append(i)
    elif i[0] == "2":
        class2.append(i)
    elif i[0] == "3":
        class3.append(i)
    elif i[0] == "4":
        class4.append(i)
    elif i[0] == "5":
        class5.append(i)
    else:
        title.append(i)
class1.insert(0,title[0]) # 把每个生成的Excel中插入title
class2.insert(0,title[0])
class3.insert(0,title[0])
class4.insert(0,title[0])
class5.insert(0,title[0])
print(class1)
# 把每个年级的数据分别保存为一个文件
def get_datas(datas,path):
    workbook = openpyxl.Workbook()
    worksheet = workbook.active
    # worksheet.title = "Class1_Datas"
    counter = 0
    for lines in datas:
        #print(lines)
        counter = counter + 1
        for i in range(len(lines)):
            worksheet.cell(counter, i + 1, lines[i])
        workbook.save(path)
    return workbook

if __name__ == "__main__":
    #设定生成文件的路径并指明文件名
    class1_datas = R"C:\Users\Administrator\python处理Excel数据\20200113\各个班级成绩单\class1_datas.xlsx"
    class2_datas = R"C:\Users\Administrator\python处理Excel数据\20200113\各个班级成绩单\class2_datas.xlsx"
    class3_datas = R"C:\Users\Administrator\python处理Excel数据\20200113\各个班级成绩单\class3_datas.xlsx"
    class4_datas = R"C:\Users\Administrator\python处理Excel数据\20200113\各个班级成绩单\class4_datas.xlsx"
    class5_datas = R"C:\Users\Administrator\python处理Excel数据\20200113\各个班级成绩单\class5_datas.xlsx"

    #调用方法写入文件
    print("班级1成绩单生成成功！",get_datas(class1,class1_datas))
    print("班级2成绩单生成成功！",get_datas(class2, class2_datas))
    print("班级3成绩单生成成功！", get_datas(class3, class3_datas))
    print("班级4成绩单生成成功！",get_datas(class4, class4_datas))
    print("班级5成绩单生成成功！",get_datas(class5, class5_datas))

 三，结果生成

图1：为调用生成的文件数量

图2：这里只打开一个文件作为例子。可以看出我们成功的把数据按照班级进行了划分