Python 如何绘制能带结构

我想画一个能带结构。积分转移矩阵（H）和重叠矩阵（S），我使用2x2阶，我可以用下面的代码绘制能带结构：

H=np.array（[[H11，H12]，[H21，H22]]
S=np.数组（[[S11，S12]，[S21，S22]]
#厄米矩阵
Ek，Cvec=直线度八（H，S）
Ec[ik]=Ek[1]
Ev[ik]=Ek[0]
#图能带结构
从matplotlib导入pyplot作为plt
图，ax=plt.子批次（）
ax.地块（ka、Ec）
ax.图（ka，Ev）

但当我想用6x6矩阵作图时，我有困难。有谁能帮我使用代码方法吗。多谢各位