import numpy from mpi4py import MPI comm = MPI.COMM_WORLD rank = comm.Get_rank() size = comm.Get_size() if (rank == 0): data = numpy.arange(10) comm.send(data[:5], dest=1, tag=13) comm.send(data[5:], dest=2, tag=14) print("Rank %d data is: "%rank, data) elif (rank == 1): data = comm.recv(source=0, tag=13) print("Rank %d Message Received, data is: "%rank, data) elif (rank == 2): data = comm.recv(source=0, tag=14) print("Rank %d Message Received, data is: "%rank, data) # end if