温馨提示×

python多进程怎么共享数据

小亿
134
2023-11-16 14:30:57
栏目: 编程语言

在Python中,多进程之间可以通过以下几种方式共享数据:

  1. 共享内存:使用multiprocessing.Valuemultiprocessing.Array可以创建共享内存的变量和数组。这些共享变量和数组可以在多个进程之间进行读写操作。
from multiprocessing import Process, Value, Array

def update_value(v):
    v.value += 1

def update_array(a):
    for i in range(len(a)):
        a[i] += 1

if __name__ == '__main__':
    v = Value('i', 0)
    a = Array('i', [0, 1, 2, 3, 4])

    p1 = Process(target=update_value, args=(v,))
    p2 = Process(target=update_array, args=(a,))

    p1.start()
    p2.start()

    p1.join()
    p2.join()

    print(v.value)  # 输出: 1
    print(a[:])  # 输出: [1, 2, 3, 4, 5]
  1. 管道(Pipe):使用multiprocessing.Pipe可以创建进程之间的管道,并通过管道发送和接收数据。
from multiprocessing import Process, Pipe

def send_data(conn):
    conn.send("Hello from child process!")

def receive_data(conn):
    data = conn.recv()
    print(data)

if __name__ == '__main__':
    parent_conn, child_conn = Pipe()

    p1 = Process(target=send_data, args=(child_conn,))
    p2 = Process(target=receive_data, args=(parent_conn,))

    p1.start()
    p2.start()

    p1.join()
    p2.join()
  1. 共享队列(Queue):使用multiprocessing.Queue可以创建进程之间的队列,并通过队列发送和接收数据。
from multiprocessing import Process, Queue

def send_data(q):
    q.put("Hello from child process!")

def receive_data(q):
    data = q.get()
    print(data)

if __name__ == '__main__':
    q = Queue()

    p1 = Process(target=send_data, args=(q,))
    p2 = Process(target=receive_data, args=(q,))

    p1.start()
    p2.start()

    p1.join()
    p2.join()

需要注意的是,由于多进程之间是独立的,所以数据的共享需要通过上述方式进行显式地操作。另外,对于一些特殊的数据类型(如字典、集合等),需要使用multiprocessing.Manager来创建共享对象。

0