Get MPI rank in the executing shell script

executable.sh

#!/bin/bash

rank=`python get_mpi_rank.py`
if [ ${rank} -eq 0 ]; then
    echo 'I am rank 0'
elif [ ${rank} -eq 1 ]; then
    echo 'I am rank 1'
elif [ ${rank} -eq 2 ]; then
    echo 'I am rank 2'
elif [ ${rank} -eq 3 ]; then
    echo 'I am rank 3'
fi

# $ mpirun -np 4 executable.sh
# I am rank 1
# I am rank 3
# I am rank 2
# I am rank 0

get_my_rank.py

from mpi4py import MPI

comm = MPI.COMM_WORLD
print(comm.Get_rank())

the file name should be get_mpi_rank not get_my_rank

Thank you so much. Before I saw your code, I have no idea how to get the current rank of the bash script. Now I know I need to run a program inside the bash script to find its rank.