00000000.py

import numpy as np
import matplotlib.pyplot as plt

# Parameters
L = 1.0          # Length of the box
n_levels = 3     # Number of energy levels to plot
x = np.linspace(0, L, 1000)  # Positions inside the box

plt.figure(figsize=(10, 6))

for n in range(1, n_levels + 1):
    # Wavefunction Ψ_n(x)
    psi_n = np.sqrt(2/L) * np.sin(n * np.pi * x / L)
    
    # Probability density |Ψ_n(x)|^2
    prob_density = psi_n**2
    
    # Plot wavefunction
    plt.plot(x, psi_n, label=f'Ψ_{n}(x)')
    
    # Optionally, plot probability density (uncomment below)
    # plt.plot(x, prob_density, linestyle='--', label=f'|Ψ_{n}(x)|^2')

plt.title("Wavefunctions of a Particle in a 1D Box")
plt.xlabel("Position x")
plt.ylabel("Ψ_n(x)")
plt.legend()
plt.grid(True)
plt.show()