calculate and plot energy band

[kaiqi-wang]

Dear zeying,
I use MagneticKP-python version to print Hamiltonian , and run the example in the file of /MagneticKP-main/MagneticKP-python/doc/html/index.html. I also want to know how to use this Hamiltonian to calculate and plot energy band ?

[zhangzeyingvv]

Hi kaiqi, You can use the matplotlib module to plot the band structure, Here is a simple example to plot the energy band structure of 2x2 model

import matplotlib.pyplot as plt

C_0_0 = 1
C_2_1 = 2

def matrix(kx):
    return np.array([
        [C_0_0, -C_2_1 * kx],
        [-C_2_1 * kx, C_0_0]
    ])

# Define the range of kx values
kx_range = np.linspace(-10, 10, 1000)

# Calculate the eigenvalues for each kx value
eigenvalues = [np.linalg.eigvals(matrix(kx_val)) for kx_val in kx_range]

# Unpack the eigenvalues into separate arrays
eval1, eval2 = zip(*eigenvalues)

# Plot the eigenvalues
plt.plot(kx_range, eval1, label="Eigenvalue 1")
plt.plot(kx_range, eval2, label="Eigenvalue 2")
plt.xlabel("kx")
plt.ylabel("Eigenvalues")
plt.legend()
plt.show()

Or you can use the Mathematica version of MagneticKP, and plot the band structure by bandManipulate and bandplot functions.

[kaiqi-wang]

Thank you very much ~ Is there a WeChat or QQ communication group? I would like to apply to join.

[zhangzeyingvv]

Yes! You can scan this QR code: