py.guanjihuan.com/Tutorial0/Fourier_transform_and_calculate_band_structures.py.py
2022-01-19 18:45:05 +08:00

9 lines
425 B
Python

import guan
import numpy as np
import functools
# Fourier transform / calculate band structures / plot figures
x_array = np.linspace(-np.pi, np.pi, 100)
hamiltonian_function = functools.partial(guan.one_dimensional_fourier_transform, unit_cell=0, hopping=1)
eigenvalue_array = guan.calculate_eigenvalue_with_one_parameter(x_array, hamiltonian_function)
guan.plot(x_array, eigenvalue_array, xlabel='k', ylabel='E', type='-k')