pymatviz

A toolkit for visualizations in materials informatics.

If you use pymatviz in your research, see how to cite.

Installation

pip install pymatviz

API Docs

See the /api page.

Usage

See the Jupyter notebooks under examples/ for how to use pymatviz. PRs with additional examples are welcome! 🙏

mlff_phonons.ipynb		Launch Codespace
matbench_dielectric_eda.ipynb		Launch Codespace
mp_bimodal_e_form.ipynb		Launch Codespace
matbench_perovskites_eda.ipynb		Launch Codespace
mprester_ptable.ipynb		Launch Codespace

Periodic Table

See pymatviz/ptable/ptable_matplotlib.py and pymatviz/ptable/ptable_plotly.py. matplotlib supports heatmaps, heatmap ratios, heatmap splits (multiple values per element), histograms, scatter plots and line plots. plotly currently only supports heatmaps (PRs to port over other matplotlib ptable variants to plotly are very welcome!). The plotly heatmap supports displaying additional data on hover or full interactivity through Dash.

ptable_heatmap(compositions, log=True)	ptable_heatmap_ratio(comps_a, comps_b)
ptable_heatmap_plotly(atomic_masses)	ptable_heatmap_plotly(compositions, log=True)
ptable_hists(data, colormap="coolwarm")	ptable_hists_plotly(data)
ptable_scatters(data, colormap="coolwarm")	ptable_lines(data)
ptable_heatmap_splits(2_vals_per_elem, colormap="coolwarm", start_angle=135)	ptable_heatmap_splits(3_vals_per_elem, colormap="coolwarm", start_angle=90)
ptable_heatmap_splits_plotly(2_vals_per_elem)	ptable_heatmap_splits_plotly(3_vals_per_elem)

Dash app using ptable_heatmap_plotly()

See examples/mprester_ptable.ipynb.

2022-07-28-ptable_heatmap_plotly-dash-example.mp4

Phonons

See examples/mlff_phonons.ipynb for usage example.

phonon_bands(bands_dict)	phonon_dos(doses_dict)
phonon_bands_and_dos(bands_dict, doses_dict)	phonon_bands_and_dos(single_bands, single_dos)

Structure

See pymatviz/structure_viz/(mpl|plotly).py. Currently structure plotting is only supported with matplotlib in 2d. 3d interactive plots (probably with plotly) are on the road map.

structure_2d(mp_19017)	structure_2d(mp_12712)
structure_2d_plotly(six_structs)	structure_3d_plotly(six_structs)

X-Ray Diffraction

See pymatviz/xrd.py.

xrd_pattern(pattern)	xrd_pattern({key1: patt1, key2: patt2})
xrd_pattern(struct_dict, stack="horizontal")	xrd_pattern(struct_dict, stack="vertical", title="Custom Title")

Radial Distribution Functions

See pymatviz/rdf/plotly.py.

rdf_plot(rdf)	rdf_plot(rdf, rdf2)

Coordination

See pymatviz/coordination/plotly.py.

coordination_hist(struct_dict)	coordination_hist(struct_dict, by_element=True)
coordination_vs_cutoff_line(struct_dict, strategy=None)	coordination_vs_cutoff_line(struct_dict, strategy=None)

Sunburst

See pymatviz/sunburst.py.

spacegroup_sunburst([65, 134, 225, ...])	spacegroup_sunburst(["C2/m", "P-43m", "Fm-3m", ...])

Rainclouds

See pymatviz/rainclouds.py.

rainclouds(two_key_dict)	rainclouds(three_key_dict)

Sankey

See pymatviz/sankey.py.

sankey_from_2_df_cols(df_perovskites)	sankey_from_2_df_cols(df_space_groups)

Histograms

See pymatviz/histogram.py.

spacegroup_bar([65, 134, 225, ...], backend="matplotlib")	spacegroup_bar(["C2/m", "P-43m", "Fm-3m", ...], backend="matplotlib")
spacegroup_bar([65, 134, 225, ...], backend="plotly")	spacegroup_bar(["C2/m", "P-43m", "Fm-3m", ...], backend="plotly")
elements_hist(compositions, log=True, bar_values='count')	histogram({'key1': values1, 'key2': values2})

Scatter Plots

See pymatviz/scatter.py.

density_scatter_plotly(df, x=x_col, y=y_col, ...)	density_scatter_plotly(df, x=x_col, y=y_col, ...)
density_scatter(xs, ys, ...)	density_scatter_with_hist(xs, ys, ...)
density_hexbin(xs, ys, ...)	density_hexbin_with_hist(xs, ys, ...)

Uncertainty

See pymatviz/uncertainty.py.

qq_gaussian(y_true, y_pred, y_std)	qq_gaussian(y_true, y_pred, y_std: dict)
error_decay_with_uncert(y_true, y_pred, y_std)	error_decay_with_uncert(y_true, y_pred, y_std: dict)

Cumulative Metrics

See pymatviz/cumulative.py.

cumulative_error(preds, targets)	cumulative_residual(preds, targets)

Classification

See pymatviz/classify/confusion_matrix.py.

confusion_matrix(conf_mat, ...)	confusion_matrix(y_true, y_pred, ...)

See pymatviz/classify/curves_plotly.py.

roc_curve_plotly(targets, probs_positive)	precision_recall_curve_plotly(targets, probs_positive)

How to cite pymatviz

See citation.cff or cite the Zenodo record using the following BibTeX entry:

@software{riebesell_pymatviz_2022,
  title = {Pymatviz: visualization toolkit for materials informatics},
  author = {Riebesell, Janosh and Yang, Haoyu and Goodall, Rhys and Baird, Sterling G.},
  date = {2022-10-01},
  year = {2022},
  doi = {10.5281/zenodo.7486816},
  url = {https://github.com/janosh/pymatviz},
  note = {10.5281/zenodo.7486816 - https://github.com/janosh/pymatviz},
  urldate = {2023-01-01}, % optional, replace with your date of access
  version = {0.8.2}, % replace with the version you use
}