Мой код для SEQ с интеграцией квантовой чехарды в Python ⇐ Python

[Anonymous]

По какой-то причине анимация не работает, я проверил кое-что в pycharm и запустил несколько примеров кодов из Интернета, так что проблема определенно в моем коде... Так может кто-нибудь сказать мне, почему анимация не работает? ?
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation

# Define constants
hbar = 1.0
m = 1.0
N = 256 # Number of grid points
L = 10.0 # Size of the spatial domain
dx = L / N
dt = 0.01 # Time step
timesteps = 200 # Number of time steps to animate

# Define spatial grid
x = np.linspace(-L/2, L/2, N)
k = 2 * np.pi * np.fft.fftfreq(N, d=dx)

# Define potential V(x)
V = 0.5 * x**2

# Initial wavefunction
psi_0 = np.exp(-x**2 / 2) # Ground state
psi_1 = x * np.exp(-x**2 / 2) # First excited state

# Choose initial state
psi = psi_0 + 1j * 0

# Time evolution using the quantum leapfrog algorithm (split-step Fourier method)
def evolve(psi, V, dt):
psi_k = np.fft.fft(psi)
psi_k *= np.exp(-1j * hbar * k**2 / (2 * m) * dt / 2)
psi = np.fft.ifft(psi_k)
psi *= np.exp(-1j * V * dt / hbar)
psi_k = np.fft.fft(psi)
psi_k *= np.exp(-1j * hbar * k**2 / (2 * m) * dt / 2)
psi = np.fft.ifft(psi_k)
return psi

# Set up the figure and axis
fig, ax = plt.subplots()
line, = ax.plot(x, np.abs(psi)**2)

def update(frame):
global psi
psi = evolve(psi, V, dt)
line.set_ydata(np.abs(psi)**2)
return line,

ani = animation.FuncAnimation(fig, update, frames=timesteps, blit=True, interval=50)
ax.set_ylim(0, 1)
plt.xlabel('x')
plt.ylabel('Probability Density |ψ|²')
plt.title('Time Evolution of the Wavefunction')

plt.show()



Подробнее здесь: https://stackoverflow.com/questions/786 ... -in-python