# 導(dǎo)入模塊
from openpyxl import load_workbook,Workbook

# 加載excel
def get_data(file):
    wb = load_workbook(file)
    ws = wb.active
    data = []
    for row in ws.iter_rows():
        data.append([cell.value for cell in row])
    wb.close()
   #注意這里很重要
    return data

def write_data(data):
    wb = Workbook()
    # 創(chuàng)建1-12月份的worksheet
    for i in range(1,13):
        ws = wb.create_sheet(f'{i}月份',i-1)
        # 添加表頭
        ws.append(data[0])
        # 判斷員工生日屬于哪一個月份
        for row in data[1:]:
            row[-1] = row[2][10:14] # 提取生日賦值給最后一個cell
            if f'{i:02}' == row[2][10:12]: # 如果員工生日是當前月份
                ws.append(row)
                # 設(shè)置排序
            ws.auto_filter.ref = ws.dimensions
            ws.auto_filter.add_sort_condition(ws.dimensions)

    wb.save('員工生日.xlsx')

if __name__ == '__main__':
    file = '../data/員工信息.xlsx'
    data = get_data(file)
    write_data(data)

說明:本文章為大熊自動化辦公課程的學習筆記