Python_mpi_rank

# run this python script using
# mpirun -np 4 this_script.py
import os

# Assuming you need to do 24 jobs, which is labeled as 0, 1, 2 ... 23
job_size = 24

# Environment Variable provide by mpirun
worker_size = int(os.environ['OMPI_COMM_WORLD_SIZE'])
worker_id = int(os.environ['OMPI_COMM_WORLD_RANK'])

all_jobs = range(24)
job_for_this_worker = [x for x in all_jobs if x % worker_size == worker_id]

print("I am worker #%d, I'm going to run "%worker_id, job_for_this_worker)

# then start run the code you want
# for x in job_for_this_worker:
#     do_something(x)