add python script for mpi plot