Python编程实践 - 多线程

简介

本文介绍在Python中如何实现多线程，包含如下要点：

1. 面向对象的继承，实现自定义的多线程类
2. 队列的使用，包括队列的初始化、赋值、取数

一、 面向对象的继承，实现自定义的多线程类

继承threading.Thread类，并实现__init__方法和run方法

class myThread (threading.Thread):
    def __init__(self, threadID, threadName, q):
        threading.Thread.__init__(self)
        self.threadID = threadID
        self.threadName = threadName
        self.q = q
    def run(self):
        print ("开启线程：" + self.name)
        self.process_data()
        print ("退出线程：" + self.name)

二、 队列的使用

队列的特性是先进先出，可保证处理的顺序性。如果没有顺序性的要求，也可以用列表、数组等来替代。

    # 队列初始化，队列大小有日期列表的长度而定
    work_queue = queue.Queue(len(date_list))
    # 队列赋值
    for word in date_list:
        work_queue.put(word)
    # 队列中取数
    data = work_queue.get()

完整代码

# coding=utf-8
import os
import string
import pandas as pd
import queue
import threading
import time

class MyThread (threading.Thread):
    def __init__(self, thread_id, thread_name, work_queue):
        threading.Thread.__init__(self)
        self.thread_id = thread_id
        self.thread_name = thread_name
        self.work_queue = work_queue
    def run(self):
        print ("开启线程：" + self.thread_name)
        self.process_data()
        print ("退出线程：" + self.thread_name)

    def process_data(self):
        while not exitFlag:
            if not self.work_queue.empty():
                ## 从队列中取出一个数据对象
                data = self.work_queue.get()
                print ("%s processing %s" % (self.thread_name, data))
            time.sleep(1)

if __name__ == '__main__':
    ## 用于标识是否结束线程，当exitFlag = 1时，结束线程
    exitFlag = 0
    thread_num = 3
    start_date = '20210101'
    end_date = '20210201'
    date_list = pd.date_range(start=start_date, end=end_date, freq="D")
    # 队列初始化，队列大小有日期列表的长度而定
    work_queue = queue.Queue(len(date_list))
    # 队列赋值
    for word in date_list:
        work_queue.put(word)


    threads = []
    # 创建新线程
    for thread_id in range(1, thread_num+1):
        thread_name = 'Thread-%d'%thread_id
        thread = MyThread(thread_id, thread_name, work_queue)
        thread.start()
        threads.append(thread)



    # 等待队列清空
    while not work_queue.empty():
        pass

    # 通知线程是时候退出
    exitFlag = 1

    # 等待所有线程完成
    for t in threads:
        t.join()
    print ("退出主线程")