update
This commit is contained in:
@@ -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()
|
||||||
|
|
||||||
|
|
||||||
|
@@ -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)
|
Reference in New Issue
Block a user