如何将来自不同等级的列表连接到等级 0 的单个列表中？

Question

我正在创建不同等级的本地列表并将它们收集到等级 0。我想创建一个主列表，其中包含按顺序包含本地列表的元素。

我目前的代码如下

from mpi4py import MPI

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

local_list = [i for i in range(rank*2, (rank+1)*2)]

master = []
master = comm.gather(local_list, root=0)
if rank == 0:
    print(master)

在使用 4 个处理器运行此代码时，我得到的输出为：

[[0, 1], [2, 3], [4, 5], [6, 7]]

但我想要的输出是：

[0, 1, 2, 3, 4, 5, 6, 7]

Answer 1

我不确定mpi4py 包的作用，但您可以使用以下代码来展平您的列表：

flat_list = [item for sublist in master for item in sublist]