【Python】處理Excel中數據2 (按照班級篩選數據_生成多個文件)

一，源數據如下

需求：1. 按照班級把同一個班級的所有成員篩選出來。

　　　2. 篩選出來的數據作為單獨的Excel表生成保存

二，代碼的編寫

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

import openpyxl

# 1.加載Excel源數據
path = R"C:\Users\Administrator\python處理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：這里只打開一個文件作為例子。可以看出我們成功的把數據按照班級進行了划分