py.guanjihuan.com/tutorial/some_models_in_the_reciprocal_space.py
guanjihuan cc1f75895c update
2021-09-24 04:21:43 +08:00

9 lines
578 B
Python

import guan
import numpy as np
# Hamiltonian of models in the reciprocal space / calculate band structures / plot figures
x_array = np.linspace(-np.pi, np.pi, 100)
eigenvalue_array = guan.calculate_eigenvalue_with_one_parameter(x_array, guan.hamiltonian_of_square_lattice_in_quasi_one_dimension)
guan.plot(x_array, eigenvalue_array, xlabel='k', ylabel='E', type='-k')
eigenvalue_array = guan.calculate_eigenvalue_with_one_parameter(x_array, guan.hamiltonian_of_graphene_with_zigzag_in_quasi_one_dimension)
guan.plot(x_array, eigenvalue_array, xlabel='k', ylabel='E', type='-k')