【Python】處理Excel中數據3 (按照班級篩選數據_把數據保存在一個文件的各個sheet中)

一，源數據的載入

需求： 1. 按照班級篩選

　　　 2. 把篩選出來的數據分別保存在一個文件的各個sheet中

 

二，代碼編寫

import pandas as pd
import os
import time
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)

#文件保存路徑
path = R"C:\\Users\\Administrator\\python處理Excel數據\\20200225"

if not os.path.exists(path):
    os.mkdir(path)
    print("目錄創建成功")
else:
    print("目錄已經存在")
# 工作表名字
date_rand = time.strftime("%Y-%m-%d", time.localtime())
file_name = path + '\%s.xlsx' % date_rand
writer = pd.ExcelWriter(file_name)

#生成新文件的各個sheet.並添加title
sheet1 = pd.DataFrame(class1,columns=["班級","姓名","語文","數學","英語","總分","性別","學號"])
sheet2 = pd.DataFrame(class2,columns=["班級","姓名","語文","數學","英語","總分","性別","學號"])
sheet3 = pd.DataFrame(class3,columns=["班級","姓名","語文","數學","英語","總分","性別","學號"])
sheet4 = pd.DataFrame(class4,columns=["班級","姓名","語文","數學","英語","總分","性別","學號"])
sheet5 = pd.DataFrame(class5,columns=["班級","姓名","語文","數學","英語","總分","性別","學號"])

#寫入數據到各個sheet中並指定列行標
sheet1.to_excel(writer,"sheet1",startcol=0,index=False)
sheet2.to_excel(writer,"sheet2",startcol=0,index=False)
sheet3.to_excel(writer,"sheet3",startcol=0,index=False)
sheet4.to_excel(writer,"sheet4",startcol=0,index=False)
sheet5.to_excel(writer,"sheet5",startcol=0,index=False)

# 將緩存寫入工作表
writer.save()
print("數據寫入成功！")

三，顯示結果