This commit is contained in:
2021-12-21 18:04:54 +08:00
parent 781854b81a
commit b39bf04dd4
2 changed files with 6 additions and 6 deletions

View File

@@ -33,8 +33,8 @@ def hamiltonian(k1, k2, M, t1, a=1/sqrt(3)): # graphene哈密顿量a为原
def main(): def main():
hamiltonian0 = functools.partial(hamiltonian, M=0, t1=1, a=1/sqrt(3)) # 使用偏函数,固定一些参数 hamiltonian0 = functools.partial(hamiltonian, M=0, t1=1, a=1/sqrt(3)) # 使用偏函数,固定一些参数
k1 = np.linspace(-2*pi, 2*pi, 800) k1 = np.linspace(-2*pi, 2*pi, 500)
k2 = np.linspace(-2*pi, 2*pi, 800) k2 = np.linspace(-2*pi, 2*pi, 500)
plot_bands_two_dimension(k1, k2, hamiltonian0) plot_bands_two_dimension(k1, k2, hamiltonian0)
@@ -57,7 +57,7 @@ def plot_bands_two_dimension(k1, k2, hamiltonian):
ax = fig.gca(projection='3d') ax = fig.gca(projection='3d')
k1, k2 = np.meshgrid(k1, k2) k1, k2 = np.meshgrid(k1, k2)
for dim0 in range(dim): for dim0 in range(dim):
ax.plot_surface(k1, k2, eigenvalue_k[:, :, dim0], cmap=cm.coolwarm, linewidth=0, antialiased=False) ax.plot_surface(k1, k2, eigenvalue_k[:, :, dim0], rcount=200, ccount=200, cmap=cm.coolwarm, linewidth=0, antialiased=False)
plt.show() plt.show()

View File

@@ -20,7 +20,7 @@ def hamiltonian(kx, ky): # kagome lattice
h = -t*h h = -t*h
return h return h
kx_array = np.linspace(-pi ,pi, 100) kx_array = np.linspace(-pi ,pi, 500)
ky_array = np.linspace(-pi ,pi, 100) ky_array = np.linspace(-pi ,pi, 500)
eigenvalue_array = guan.calculate_eigenvalue_with_two_parameters(kx_array, ky_array, hamiltonian) eigenvalue_array = guan.calculate_eigenvalue_with_two_parameters(kx_array, ky_array, hamiltonian)
guan.plot_3d_surface(kx_array, ky_array, eigenvalue_array, xlabel='kx', ylabel='ky', zlabel='E') guan.plot_3d_surface(kx_array, ky_array, eigenvalue_array, xlabel='kx', ylabel='ky', zlabel='E', rcount=200, ccount=200)