pymatgen绘制能带态密度图（band+DOS）

现在做材料数据挖掘都应该用Pymatgen了吧，功能十分强大。我才刚入门，做做笔记。pymatgen是一个API包，可以连接materials project进行高通量计算。现在我用pymatgen画一下能带+DOS图，炒鸡好看。

安装请见https://pypi.org/project/pymatgen/

前提是需要有vasprun.xml，这是VASP运行时就会生成的文件，小程序：

import matplotlib.pyplot as plt
import matplotlib as mpl
mpl.use('Agg')
from pymatgen.io.vasp.outputs import Vasprun
from pymatgen.electronic_structure.plotter import BSDOSPlotter,\
BSPlotter,BSPlotterProjected,DosPlotter

# read vasprun.xml，get band and dos information
bs_vasprun = Vasprun("vasprun.xml",parse_projected_eigen=True)
bs_data = bs_vasprun.get_band_structure(line_mode=True)

dos_vasprun=Vasprun("vasprun.xml")
dos_data=dos_vasprun.complete_dos

# set figure parameters, draw figure
banddos_fig = BSDOSPlotter(bs_projection='elements', dos_projection='elements',
vb_energy_range=4, fixed_cb_energy=4)
banddos_fig.get_plot(bs=bs_data, dos=dos_data)
plt.savefig('banddos_fig.png')

如下:

注意了，这里bs_projection和dos_projection的参数都是element，是包含了元素信息的，另外DOS图是有阴影填充的，这在INCAR中要设置 LORBIT >= 10，才有投影信息，建议设置成LORBIT = 11。

BSDOSPlotter的参数如下：

bs_projection (str) – “elements” or None

dos_projection (str) – “elements”, “orbitals”, or None，元素或轨道

vb_energy_range (float) – energy in eV to show of valence bands

cb_energy_range (float) – energy in eV to show of conduction bands

fixed_cb_energy (bool) – If true, the cb_energy_range will be interpreted as constant (i.e., no gap correction for cb energy)

egrid_interval (float) – interval for grid marks

font (str) – font family

axis_fontsize (float) – font size for axis

tick_fontsize (float) – font size for axis tick labels

legend_fontsize (float) – font size for legends

bs_legend (str) – matplotlib string location for legend or None

dos_legend (str) – matplotlib string location for legend or None

rgb_legend (bool) – (T/F) whether to draw RGB triangle/bar for element proj.

fig_size (tuple) – dimensions of figure size (width, height)