学习python多进程时需要实现两个进程对一个队列进行读或者写,错误代码如下:

import time
from multiprocessing import Process
from queue import Queue
import random

list1 = [1,2,3,4,5]

# 写数据进程代码
def write(queue):

    for i in list1:
        queue.put(i)
        print("存放数据",i)
        time.sleep(random.random()) #

# 读数据的代码
def read(queue):
    while True:
        if not queue.empty():
            print("获取数据:",queue.get())
            time.sleep(random.random()) #
        else:
            break

if __name__ == '__main__':
    queue = Queue(5)
    p1 = Process(target=write,args=(queue,))
    p2 = Process(target=read,args=(queue,))
    p1.start()
    p1.join()
    p2.start()
    p2.join()

运行结果为:

出现问题的原因在Queue对象导包

队列有两类:

torch.multiprocessing.Queue
queue.Queue

前者用在多进程里,后者用在多线程中。

一个是对应进程的队列对象,而一个是对于线程的对象

因此这里误导入了线程包中的queue对象出现了以上情况

解决方式,修改导入包即可

正确代码如下:

import time
from multiprocessing import Process
from multiprocessing import Queue
import random

list1 = [1,2,3,4,5]

# 写数据进程代码
def write(queue):

    for i in list1:
        queue.put(i)
        print("存放数据",i)
        time.sleep(random.random()) #

# 读数据的代码
def read(queue):
    while True:
        if not queue.empty():
            print("获取数据:",queue.get())
            time.sleep(random.random()) #
        else:
            break

if __name__ == '__main__':
    queue = Queue(5)
    p1 = Process(target=write,args=(queue,))
    p2 = Process(target=read,args=(queue,))
    p1.start()
    p1.join()
    p2.start()
    p2.join()

运行结果:

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐