import mpi if mpi.rank == 0: print "size=",mpi.size