How to visualize parallel grid partitioning？

General
1

Dear all!
I want to visually view the partitioning of the grid when running parallel programs.I couldn’t find any examples.

For example, for a rectangular grid：

from mpi4py import MPI
import numpy as np
from petsc4py import PETSc
import ufl 
from dolfinx import default_scalar_type, fem, la 
from dolfinx.cpp.fem import compute_integration_domains 
from dolfinx.mesh import (
    GhostMode,
    compute_incident_entities,
    create_box,
    create_rectangle,
    create_submesh,
    create_unit_cube,
    create_unit_interval,
    create_unit_square,
    locate_entities,
    locate_entities_boundary,
    meshtags,
)

comm = MPI.COMM_WORLD 
msh = create_unit_square(comm, 20,  20, ghost_mode=GhostMode.none,cell_type=mesh.CellType.triangle)
2

See Visualisation of distributed mesh