python基础学习（一），操作Excel文件，处理数据

基础认知（办公自动化场景）
目标：熟悉变量、条件、循环、函数，摆脱“手工作业”。

场景：公司HR需要从一份混乱的Excel考勤表中，自动筛选出迟到超过3次的员工名单。

具体例子：读取CSV文件，遍历行，统计迟到次数，输出预警名单。

练习：写一个函数 count_late(records)，输入列表，返回字典。

 

基本涉及模块：

1）使用import 进行插件引用（如果是内置插件，可以直接引用；如果是三方插件，则需要pip install 安装后，再进行引用；）

2）定义子函数，根据业务功能需求，定义具体的实现子函数；比如当前项目，可以单独定义一个获取csv文件的函数

3）主函数main();整个py文件的执行主体；定义整个业务流程；

4）定义py文件的标准入口：

if __name__ == "__main__":
main()

5）终端中，使用python late_warning.py  , 即可实现需要的业务功能；

 

以下是具体代码，可以直接使用；

具体实现代码：late_warning.py

import csv  #引用插件

def em_latecount(myrecords):
    """ 统计员工列表中，每个员工的迟到次数
    参数：myrecords 属于一个列表数组，每个元素是一个JSON对象，包含姓名和是否迟到信息
    
    返回数据：JSON数组，里面有姓名和迟到次数
    """
    latecount = {}
    for record in myrecords:
        name = record["员工姓名"]
        is_late = record["是否迟到"]
        if is_late == "是":
            if name in latecount:
                latecount[name] +=1
            else:
                latecount[name] = 1

    return latecount

def read_csv(file):
    """ 读取Excel文件，获取员工迟到记录数据表格

    参数：csv文件的路径
    
    返回：作为一个list，返回；
    """
    records = []
    with open(file,"r",encoding="utf-8") as ff:
        reader = csv.DictReader(ff)
        for row in reader:
            records.append(row)
    return records

def search_warning_list(latecount,threshold=4):
    """查找列表中，迟到次数超过4次的员工
    参数：需要查找的对象列表，及设置的迟到次数的阈值
    返回：迟到4次以上的员工列表
    """

    warning_record = []
    for name, count in latecount.items():
        if count > threshold:
            warning_record.append({"name":name,"count":count})
    return warning_record

def main():
    import os 
    filepath= os.path.dirname(os.path.abspath(__file__))
    print(f"获取csv数据路径:{filepath}")
    csvpath = os.path.join(filepath,"attendance.csv")
    
    print(f"获取csv数据:{csvpath}")
    records = read_csv(csvpath)
    print(f"获取了{len(records)}条记录\n")

    print("开始统计员工的迟到次数")
    late_count = em_latecount(records)
    
    print("打印员工迟到次数记录")
    for name,count in late_count.items():
        print(f"name:{name}, count:{count}")

    print("获取达到阈值的员工")
    warning_record = search_warning_list(late_count,4)

    print("打印警告列表")
    if warning_record:
        for aa in warning_record:
            print(f"name:{aa['name']},count:{aa['count']}")
    else:
        print("所有员工都未超出4次")

if __name__ == "__main__":
    main()

 测试使用附件：https://files.cnblogs.com/files/violetlength/attendance.zip?t=1787886060&download=true