如何使用mpi4py迭代其大小大于处理器数量的对象列表

from mpi4py import MPI

comm = MPI.COMM_WORLD

size = comm.Get_size()
rank = comm.Get_rank()

objects = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

chunk_size = len(objects) // size
local_objects = comm.scatter(objects, root=0)

for obj in local_objects:
    # 进行相应的操作
    print(obj)

results = comm.gather(local_results, root=0)

from mpi4py import MPI

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

objects = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

chunk_size = len(objects) // size
local_objects = comm.scatter(objects, root=0)

local_results = []
for obj in local_objects:
    # 进行相应的操作
    local_results.append(obj * 2)

results = comm.gather(local_results, root=0)

if rank == 0:
    print("Final Results:", results)