【python随笔】之【多线程并发统计多个cvs文件行数】

import threading
import csv
import os


class MyThreading(threading.Thread):
    def __init__(self, path):
        super().__init__()
        # 路径
        self.path = path
        # 行数
        self.line = -1

    def run(self):
        # 读取文件
        reader = csv.reader(open(self.path, 'r'))
        count_lines = 0
        for item in reader:
            count_lines += 1
        self.line = count_lines


if __name__ == '__main__':
    dir_path = r'00test'  # 要统计的所有文件的目录
    file_list = os.listdir(dir_path)  # 所有文件列表
    threading_list = []  # 线程列表

    # 遍历所有文件列表
    for file_name in file_list:
        # 每个文件的路径
        file_path = dir_path + '\\' + file_name
        # 创建读取这个文件行数的线程
        thread = MyThreading(file_path)
        thread.start()  # 线程开始统计
        threading_list.append(thread)  # 将此线程加入线程列表
    # 等待
    for th in threading_list:
        th.join()
    # 将统计个数装入列表
    line_list = []
    for th in threading_list:
        line_list.append(th.line)
    # 打印统计列表
    print(line_list)