EPWpy.plotting.plot_polaron_matplotlib#

Functions

plot_psir_plrn_matplotlib(Data, *[, view, ...])

Matplotlib implementation of your psir_plrn visualization.
EPWpy.plotting.plot_polaron_matplotlib.plot_psir_plrn_matplotlib(Data, *, view={}, iso_level=None, iso_alpha=0.35, iso_color=(0.9, 0.8, 0.0), point_size=40.0, cmap='viridis', quiver_length=1.0, quiver_normalize=False, figsize=(9, 8), show=True, bond_size=2.0)[source]#

Matplotlib implementation of your psir_plrn visualization.

Return type:

Tuple[Figure, Axes]

Expected keys in Data:

  • 'mat' : list/array of labels (used to build matn coloring)

  • 'x','y','z' : 1D arrays of atom (or node) coordinates (same length)

  • 'u','v','w' : 1D arrays for vectors (same length as x,y,z)

  • 'Dense' (unused by the original display; included for compatibility)

  • 'Grid'4D array (..., 3) with real-space coordinates per grid point

    e.g. Grid[ix,iy,iz,0/1/2] = x/y/z

  • 'pts' : (N,3) points (used by Mayavi's StructuredGrid) – optional here

  • 'Density' : 3D scalar field matching Grid[...,0].shape

  • 'connections' : list/array of (i,j) pairs or (i,j,dx,dy,dz)

  • optional 'in_notebook', 'backend' (ignored here)

Returns (fig, ax).