Update bands_of_SSH_model_with_two_kinds_of_fourier_transform.py

This commit is contained in:
guanjihuan 2021-08-07 19:09:21 +08:00
parent 46dda84c9e
commit b101731c06

View File

@ -5,7 +5,7 @@ import guan
v=0.6
w=1
k = np.linspace(-pi ,pi, 100)
k_array = np.linspace(-pi ,pi, 100)
def hamiltonian_1(k):
matrix = np.zeros((2, 2), dtype=complex)
@ -19,8 +19,8 @@ def hamiltonian_2(k):
matrix[1,0] = v*cmath.exp(-1j*k/2)+w*cmath.exp(1j*k/2)
return matrix
E_1 = guan.calculate_eigenvalue_with_one_parameter(k, hamiltonian_1)
guan.plot(k, E_1, xlabel='k', ylabel='E_1')
E_1_array = guan.calculate_eigenvalue_with_one_parameter(k_array, hamiltonian_1)
guan.plot(k_array, E_1_array, xlabel='k', ylabel='E_1')
E_2 = guan.calculate_eigenvalue_with_one_parameter(k, hamiltonian_2)
guan.plot(k, E_2, xlabel='k', ylabel='E_2')
E_2_array = guan.calculate_eigenvalue_with_one_parameter(k_array, hamiltonian_2)
guan.plot(k_array, E_2_array, xlabel='k', ylabel='E_2')