MPI4Py: OpenMPI如何跨进程更新字典？

from mpi4py import MPI

comm = MPI.COMM_WORLD
rank = comm.Get_rank()

# 定义一个要更新的字典
my_dict = {'key1': 'value1', 'key2': 'value2'}

# 将字典发送给其他进程
if rank == 0:
    for i in range(1, comm.Get_size()):
        comm.send(my_dict, dest=i, tag=0)
else:
    # 接收字典
    received_dict = comm.recv(source=0, tag=0)
    
    # 在接收到的字典中更新值
    received_dict['key1'] = 'new_value1'
    
    # 将更新后的字典发送回进程0
    comm.send(received_dict, dest=0, tag=1)

# 进程0接收更新后的字典
if rank == 0:
    for i in range(1, comm.Get_size()):
        updated_dict = comm.recv(source=i, tag=1)
        my_dict.update(updated_dict)

# 打印最终更新后的字典
print("Final dictionary:", my_dict)