Overlayed subplots from BSPlotterProjected.get_projected_plots_dots() and get_projected_plots_dots_patom_pmorb()

[HERTZCAI]

Python version

3.11.7

Pymatgen version

2024.4.13

Operating system version

Ubuntu 22.04.3 LTS

Current behavior

I try to get band structures with specific elements and orbitals with get_projected_plots_dots(). While a figure with 3 subfigures is expected, the program returns a figure with only one subfigure and the place where the other two subfigures are supposed to be is blank.

However, when I downgrade pymatgen to 2023.8.10, no such problems occur.
get_projected_plots_dots_patom_pmorb() also does not work properly.
Moreover, there are extra axes in the figure.
matplotlib version is 3.8.0.

Expected Behavior

A figure with several subfigures displays normally

Minimal example

vasprun = Vasprun("vasprun.xml",parse_projected_eigen=True)
bs_data = vasprun.get_band_structure(line_mode=True)
plt_3=BSPlotterProjected(bs=bs_data)
plt_3.get_projected_plots_dots({'K':['s'],'Ti':['d'],'O':['p']})
plt.show()

Relevant files to reproduce this bug

No response

[DanielYang59]

Hopefully the fix produces satisfactory results for you @HERTZCAI (feel free to ping me if you have more suggestions).

Also to improve the overlapping markers, you could now pass a marker_size arg for example:

plotter = BSPlotterProjected(bs=bs_data)
plotter.get_projected_plots_dots({"Cu":["s", "d"], "O":["p"]}, marker_size=6.5)

plt.show()