Python 畫沿着高對稱性的能帶圖

將能帶圖沿着高對稱性處理爲二維數組形式，然後依次畫點連線，爲了美觀，而需要自己加線

import numpy as np
import matplotlib.pylab as plt

eig_vbm= -1.3286 # fermi energy level in scf output for metals
eig = np.load('eig.npy')
eig = eig - eig_vbm
nks = 111
nbnd = 25
ymin=-80
ymax=10
lw=0.5 # line width

for i in range(nbnd):
    line1=plt.plot(np.arange(0,nks), eig[:,i],color='grey',linewidth=lw)

# 只畫第15條
#line1=plt.plot(np.arange(0,nks), eig[:,14]-eig_vbm,color='grey',linewidth=lw)
vline=[0,40,70,110]
#vline=dline
for i in range(len(vline)):
    plt.axvline(x=vline[i], ymin=ymin, ymax=ymax,linewidth=lw,color='black')
plt.xlim([0,nks-1]) # 201 points
plt.ylim([ymin,ymax])

plt.ylabel(r'E (eV)',fontsize=16)
plt.xticks((0,40,70,110), (r'${\Gamma}$', 'M', 'K', r'${\Gamma}$'))

plt.ylim([-7,3])
plt.show()