笔记大全

Python如何获取多进程执行的返回值

更新时间：2023-10-16

多进程执行的返回值是什么？

在Python中，我们可以使用多进程并行地执行某个任务。但是，当我们的任务结束后，我们如何获取多进程执行的返回值呢？答案是使用Queue对象。我们可以在主进程中创建一个队列，然后将子进程的返回值放入队列中，主进程就可以通过队列获取子进程的返回值。

from multiprocessing import Process, Queue

def worker(num, queue):
    data = num * 2
    queue.put(data)

if __name__ == '__main__':
    queue = Queue()
    processes = []
    for i in range(5):
        p = Process(target=worker, args=(i, queue))
        p.start()
        processes.append(p)
    for p in processes:
        p.join()
    results = []
    while not queue.empty():
        result = queue.get()
        results.append(result)
    print(results)

上面的代码中，我们创建了一个worker函数来模拟一个耗时操作，并将结果放入队列中。然后我们创建了5个子进程来并行地执行这个函数。在主进程中等待所有子进程执行结束，并将每个子进程的结果放入队列中。之后，我们可以通过调用队列的get()方法来获取每个子进程的结果。

如何处理多个进程的返回值？

如果我们启动了多个子进程，并且每个子进程返回多个值，该怎么办呢？我们可以使用元组来保存每个子进程的多个返回值，并将这些元组放入队列中进行传输。

from multiprocessing import Process, Queue

def worker(num, queue):
    data = (num, num * 2, num * 3)
    queue.put(data)

if __name__ == '__main__':
    queue = Queue()
    processes = []
    for i in range(5):
        p = Process(target=worker, args=(i, queue))
        p.start()
        processes.append(p)
    for p in processes:
        p.join()
    results = []
    while not queue.empty():
        result = queue.get()
        results.append(result)
    print(results)

上面的代码中，我们将每个子进程的返回值保存在一个元组中，并将这个元组放入队列中。在主进程中，我们可以通过遍历队列获取每个子进程的返回值。

如何处理多个进程的异常？

在多进程执行的过程中，子进程可能会抛出异常。如果我们没有处理这些异常，就会导致程序崩溃。所以我们需要对异常进行处理。我们可以在处理每个子进程的返回值时，检查这个返回值是否是一个异常，并对这个异常进行处理。

from multiprocessing import Process, Queue

def worker(num, queue):
    try:
        data = num * 2
        queue.put(data)
    except Exception as e:
        queue.put(e)

if __name__ == '__main__':
    queue = Queue()
    processes = []
    for i in range(5):
        p = Process(target=worker, args=(i, queue))
        p.start()
        processes.append(p)
    for p in processes:
        p.join()
    results = []
    while not queue.empty():
        result = queue.get()
        if isinstance(result, Exception):
            print('An error occurred:', result)
        else:
            results.append(result)
    print(results)

上面的代码中，我们在worker函数中使用try/except语句来捕获异常，并将异常放入队列中。在主进程中，我们在获取每个子进程的返回值时，判断这个返回值是否是一个异常，如果是，则打印异常信息。否则，我们将这个返回值放入结果列表中。

总结

在Python中，我们可以使用多进程并行地执行某个任务。当子进程执行完后，我们可以使用Queue对象获取子进程的返回值。如果有多个子进程返回多个值，我们可以将返回值保存在元组中，并将这些元组放入队列中进行传输。如果子进程在执行过程中抛出了异常，我们需要在主进程中对异常进行处理。

c语言编程笔录