guanjihuan/py.guanjihuan.com
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

guan-0.0.30

Browse Source
This commit is contained in:
guanjihuan
2021-11-16 03:46:59 +08:00
parent c5906e7e6d
commit c7cb798536
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
PyPI/src/guan/calculate_band_structures_and_wave_functions.py
View File

@@ -28,8 +28,8 @@ def calculate_eigenvalue_with_one_parameter(x_array, hamiltonian_function):
eigenvalue_array = np.zeros((dim_x, dim))
for x0 in x_array:
hamiltonian = hamiltonian_function(x0)
eigenvalue, eigenvector = np.linalg.eig(hamiltonian)
eigenvalue_array[i0, :] = np.sort(np.real(eigenvalue[:]))
eigenvalue, eigenvector = np.linalg.eigh(hamiltonian)
eigenvalue_array[i0, :] = eigenvalue
i0 += 1
return eigenvalue_array
@@ -54,8 +54,8 @@ def calculate_eigenvalue_with_two_parameters(x_array, y_array, hamiltonian_funct
j0 = 0
for x0 in x_array:
hamiltonian = hamiltonian_function(x0, y0)
eigenvalue, eigenvector = np.linalg.eig(hamiltonian)
eigenvalue_array[i0, j0, :] = np.sort(np.real(eigenvalue[:]))
eigenvalue, eigenvector = np.linalg.eigh(hamiltonian)
eigenvalue_array[i0, j0, :] = eigenvalue
j0 += 1
i0 += 1
return eigenvalue_array