Затухающий пружинный осциллятор: аналитические и численные решения на Python

Этот пост демонстрирует два подхода к моделированию затухающего пружинного осциллятора: аналитическое и численное решения.

Методы моделирования

Аналитическое решение

Затухающий гармонический осциллятор описывается дифференциальным уравнением:

$$\frac{d^2x}{dt^2} + 2\zeta\omega_0\frac{dx}{dt} + \omega_0^2x = 0$$

Для недодемпфированной системы ($\zeta < 1$) аналитическое решение:

$$x(t) = e^{-\zeta\omega_0 t}\left[x_0\cos(\omega_d t) + \frac{v_0 + \zeta\omega_0 x_0}{\omega_d}\sin(\omega_d t)\right]$$

Где:

Численное решение

Численный подход использует метод Рунге-Кутты 4-го порядка для интегрирования системы дифференциальных уравнений первого порядка:

$$\frac{dx}{dt} = v$$

$$\frac{dv}{dt} = -\omega_0^2x - 2\zeta\omega_0v$$

Этот метод обеспечивает высокую точность и позволяет сравнить с аналитическим решением для проверки корректности.

Затухающий пружинный осциллятор

Исходный код

damped_spring_oscillator.py
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# SPDX-License-Identifier: CC0 1.0 Universal

import numpy as np
import matplotlib.pyplot as plt
plt.style.use("ggplot")

# Параметры моделирования
fs = 1000  # Частота дискретизации (Гц)
t_duration = 5  # Длительность (секунды)
f_resonant = 4  # Резонансная частота (Гц)
x0 = 1.0  # Начальное смещение
v0 = 0.0  # Начальная скорость

# Производные параметры
omega0 = 2 * np.pi * f_resonant  # Собственная угловая частота
dt = 1.0 / fs  # Шаг времени
t = np.arange(0, t_duration, dt)  # Массив времени
n_samples = len(t)

# Коэффициент затухания (можно настроить для управления затуханием)
# zeta < 1: недодемпфированный, zeta = 1: критически задемпфированный, zeta > 1: передемпфированный
zeta = 0.03  # Лёгкое затухание для колебательного движения
gamma = 2 * zeta * omega0  # Коэффициент затухания

# Аналитическое решение для недодемпфированного осциллятора
omega_d = omega0 * np.sqrt(1 - zeta**2)  # Затухающая частота

# Для начальных условий x(0) = x0, v(0) = v0
# Решение: x(t) = e^(-ζω₀t)[x0*cos(ωd*t) + ((v0 + ζω₀x0)/ωd)*sin(ωd*t)]
x_analytical = np.exp(-zeta * omega0 * t) * (
    x0 * np.cos(omega_d * t) +
    ((v0 + zeta * omega0 * x0) / omega_d) * np.sin(omega_d * t)
)

print(f"Natural frequency: {f_resonant:.2f} Hz")
print(f"Damped frequency: {omega_d / (2 * np.pi):.2f} Hz")
print(f"Damping ratio: {zeta:.2f}")
print(f"Damping coefficient gamma: {gamma:.3f}")
print(f"Natural angular frequency ω₀: {omega0:.3f} rad/s")

# Численное моделирование с использованием метода Рунге-Кутты
def spring_ode(state, t):
    """
    Дифференциальное уравнение для затухающего гармонического осциллятора
    d²x/dt² + 2ζω₀(dx/dt) + ω₀²x = 0
    state = [position, velocity]
    returns [velocity, acceleration]
    """
    x, v = state
    # acceleration = -ω₀²x - 2ζω₀v
    acceleration = -omega0**2 * x - 2 * zeta * omega0 * v
    return np.array([v, acceleration])

# Инициализация массивов для численного решения
x_numerical = np.zeros(n_samples)
v_numerical = np.zeros(n_samples)

# Начальные условия
x_numerical[0] = x0
v_numerical[0] = v0

# Интегрирование методом Рунге-Кутты 4-го порядка
for i in range(n_samples - 1):
    state = np.array([x_numerical[i], v_numerical[i]])

    k1 = dt * spring_ode(state, t[i])
    k2 = dt * spring_ode(state + k1/2, t[i] + dt/2)
    k3 = dt * spring_ode(state + k2/2, t[i] + dt/2)
    k4 = dt * spring_ode(state + k3, t[i] + dt)

    next_state = state + (k1 + 2*k2 + 2*k3 + k4) / 6
    x_numerical[i+1] = next_state[0]
    v_numerical[i+1] = next_state[1]

# Построение графиков результатов
plt.figure(figsize=(12, 8))

# График положения
plt.subplot(2, 1, 1)
plt.plot(t, x_analytical, 'b-', label='Analytical Solution', linewidth=2)
plt.plot(t[::10], x_numerical[::10], 'ro', markersize=2, label='Numerical Solution', alpha=0.7)
plt.xlabel('Time (s)')
plt.ylabel('Position (m)')
plt.title(f'Dampened Spring Oscillator (f₀={f_resonant}Hz, ζ={zeta}, fs={fs}Hz)')
plt.legend()
plt.grid(True, alpha=0.3)

# Аналитическая скорость (производная положения)
v_analytical = np.exp(-zeta * omega0 * t) * (
    (-zeta * omega0 * x0 + (v0 + zeta * omega0 * x0) * omega_d / omega_d) * np.cos(omega_d * t) +
    (-omega_d * x0 - zeta * omega0 * (v0 + zeta * omega0 * x0) / omega_d) * np.sin(omega_d * t)
)

# График скорости
plt.subplot(2, 1, 2)
plt.plot(t, v_analytical, 'g-', label='Analytical Velocity', linewidth=2)
plt.plot(t[::10], v_numerical[::10], 'mo', markersize=2, label='Numerical Velocity', alpha=0.7)
plt.xlabel('Time (s)')
plt.ylabel('Velocity (m/s)')
plt.title('Velocity of the Spring Mass')
plt.legend()
plt.grid(True, alpha=0.3)

plt.tight_layout()
plt.show()

# Вычисление ошибки между аналитическим и численным решениями
error = np.abs(x_analytical - x_numerical)
max_error = np.max(error)
rms_error = np.sqrt(np.mean(error**2))

print(f"\nNumerical accuracy:")
print(f"Maximum error: {max_error:.6f} m")
print(f"RMS error: {rms_error:.6f} m")
print(f"Relative RMS error: {rms_error/np.max(np.abs(x_analytical))*100:.4f}%")
plt.savefig("dampened_spring_oscillator.svg", dpi=300, bbox_inches='tight')

Check out similar posts by category: Python, Physics