OrderedDict在使用Mpi4py时会丢失其序列

Question

我正在尝试使用Mpi4py打印词典：

from mpi4py import MPI

comm = MPI.COMM_WORLD
rank = comm.rank

if rank == 0:
    data = {'a':1,'b':2,'c':3}
else:
    data = None

data = comm.bcast(data, root=0)
print 'rank',rank,data

一旦我使用下面的命令为更多处理器运行它：

mpiexec -n 10 python code.py

结果与其他处理器结果混合，如下所示：

rank 0 {'arank 2 {'a': 1, 'c': 3, 'b': 2}
rank' 3 {'a': 1, 'c': 3, 'b': 2}
: 1, 'c': 3, 'b': 2}
rank 8 {'a': 1, 'c': 3, 'b': 2}rank
 1 {'a': 1, 'c': 3, 'b': 2}
rank 4 {'a': 1, 'c': 3, 'b': 2}
rank 5 {'a': 1, 'c': 3, 'b': 2}
rank 9 {'a': 1, 'c': 3, 'b': 2}
rankrank  7 {'a': 1, 'c': 3, 'b': 2}
6 {'a': 1, 'c': 3, 'b': 2}

我假设发生这种情况是因为当处理器完成任务时，它们会打印结果，从而导致混合匹配。我试过使用OrderedDict，但即使这样也没用。代码如下

from mpi4py import MPI
import collections 

comm = MPI.COMM_WORLD
rank = comm.rank

if rank == 0:
    data = collections.OrderedDict({'a':1,'b':2,'c':3})
else:
    data = None

data = comm.bcast(data, root=0)
print 'rank',rank,data

有没有办法让结果保持一致，而不是和其他处理器的结果混在一起？你知道吗

Answer 1

我不确定它是否解决了混合输出的问题，但在声明中

collections.OrderedDict({'a':1,'b':2,'c':3})

有两种情况发生：

1. 创建dict文字{'a':1,'b':2,'c':3}。你知道吗
2. 将其传递给collections.OrderedDict，结果是以继承自原始dict文本的顺序生成副本。你知道吗

如果要保持插入顺序，请传递键-值对的iterable（如list）。你知道吗

所以我们可以这样写

collections.OrderedDict([('a', 1), ('b', 2), ('c', 3)])

试验

>>> data = collections.OrderedDict([('a', 1), ('b', 2), ('c', 3)])
>>> list(data.keys()) == ['a', 'b', 'c']
True