Решение нелинейной системы ОДУ с краевыми условиями на Python

Вопрос или проблема

Я пытаюсь решить эту систему дифференциальных уравнений:

введите описание изображения здесь

Функции должны вести себя следующим образом, когда r-> бесконечность (F = g = lambda = const.):

введите описание изображения здесь

Недавно у меня была похожая проблема, и один пользователь любезно предоставил мне алгоритм, с помощью которого я мог решить это ОДС. Теперь я попытался применить алгоритм TDM к новой задаче, но получил только тривиальное решение. Кто-нибудь знает, что я сделал не так, или как я могу решить эту проблему?

import numpy as np
import matplotlib.pyplot as plt

# Функция TDMA для решения уравнения трёхдиагональной матрицы
def tdma(a, b, c, d):
    n = len(d)
    p = np.zeros(n)
    q = np.zeros(n)
    x = np.zeros(n)

    # Прямой проход
    i = 0
    denominator = b[i]
    p[i] = -c[i] / denominator
    q[i] = d[i] / denominator
    for i in range(1, n):
        denominator = b[i] + a[i] * p[i - 1]
        if abs(denominator) < 1.0e-10:
            print("Нет решения")
            return x
        p[i] = -c[i] / denominator
        q[i] = (d[i] - a[i] * q[i - 1]) / denominator

    # Обратный проход
    x[n - 1] = q[n - 1]
    for i in range(n - 2, -1, -1):
        x[i] = p[i] * x[i + 1] + q[i]

    return x

# Параметры
N = 1000  # Количество ячеек (1-N); добавлены границы 0 и N+1
rmin, rmax = 0.0, 100.0  # Минимальное и максимальное r
dr = rmax / N  # Ширина ячейки
lamda = 10
g = 1.0  # Вы можете настроить этот параметр, если необходимо

# Граничные условия на F и W
FL, FR = 0, 1
WL, WR = 0, 1 / rmax

# Расположение ячеек / узлов
r = np.linspace(rmin - 0.5 * dr, rmax + 0.5 * dr, N + 2)
r[0], r[N + 1] = rmin, rmax
rv = np.linspace(rmin, rmax, N + 1)

# Коэффициенты
awF = np.zeros(N + 2)
aeF = np.zeros(N + 2)
awW = np.zeros(N + 2)
aeW = np.zeros(N + 2)

# Основные ячейки
aeF[1:N + 1] = 1 / dr ** 2
aeW[1:N + 1] = rv[1:N + 1] ** 2 / dr ** 2
awF[1:N + 1] = aeF[0:N]
awW[1:N + 1] = aeW[0:N]

# Границы
awF[1], awW[1], aeF[N], aeW[N] = 0, 0, 0, 0
awFL = 2 / dr ** 2
aeFR = 2 / dr ** 2
awWL = 2 * rmin ** 2 / dr ** 2
aeWR = 2 * rmax ** 2 / dr ** 2

# Начальные значения
F = np.ones(N + 2) * 1e-5
W = np.ones(N + 2) * 1e-5

# Граничные условия
F[0], F[N + 1] = FL, FR
W[0], W[N + 1] = WL, WR

# Фактор недоразмещения
alpha = 0.5
niter = 0
s = np.zeros(N + 2)
sp = np.zeros(N + 2)

# Цикл итераций
for _ in range(2000):
    niter += 1
    F_target = F.copy()
    W_target = W.copy()

    ######################
    # Обновление уравнения W #
    ######################
    for i in range(1, N + 1):
        K = 1 - g * r[i] * W[i]
        H = g * r[i] * F[i]

        # Ограничение значений для предотвращения переполнения
        K = np.clip(K, -1e5, 1e5)
        H = np.clip(H, -1e5, 1e5)

        s[i] = K * (K**2 - 1) + H**2 * K
        sp[i] = -2 * (1 - g * r[i]) / r[i]**2

        if np.isnan(K) or np.isnan(H):
            print(f"Числовая проблема при r = {r[i]}, K = {K}, H = {H}")
            break

    s[1] += awFL * FL
    sp[1] -= awFL
    s[N] += aeFR * FR
    sp[N] -= aeFR

    ap = awF + aeF - sp
    F_target[1:N + 1] = tdma(-awF[1:N + 1], ap[1:N + 1], -aeF[1:N + 1], s[1:N + 1])

    ######################
    # Обновление уравнения F #
    ######################
    for i in range(1, N + 1):
        K = 1 - g * r[i] * W[i]
        H = g * r[i] * F[i]

        # Ограничение значений для предотвращения переполнения
        K = np.clip(K, -1e5, 1e5)
        H = np.clip(H, -1e5, 1e5)

        s[i] = 2 * H * K**2 + lamda * H * ((H**2 / g**2) - r[i]**2 * F[i]**2)

        if np.isnan(K) or np.isnan(H):
            print(f"Числовая проблема при r = {r[i]}, K = {K}, H = {H}")
            break

    s[1] += awWL * WL
    sp[1] -= awWL
    s[N] += aeWR * WR
    sp[N] -= aeWR

    ap = awW + aeW - sp
    W_target[1:N + 1] = tdma(-awW[1:N + 1], ap[1:N + 1], -aeW[1:N + 1], s[1:N + 1])

    # Вычислить изменение
    change = (np.linalg.norm(F_target - F) + np.linalg.norm(W_target - W)) / N

    # Недоразмещение обновления
    F = alpha * F_target + (1 - alpha) * F
    W = alpha * W_target + (1 - alpha) * W

    if change < 1.0e-10:
        break

print(niter, " итераций ")

# Построить результаты
plt.plot(r, F, label="F(r)", color="g")
plt.plot(r, W, label="W(r)", color="r")
plt.legend(loc="upper right")
plt.xlabel("r")
plt.ylim(0)
plt.xlim(0, 20)
plt.grid(True)
plt.show()

Ответ или решение

Для решения системы нелинейных обыкновенных дифференциальных уравнений (ОДЕ) с границами в Python, необходимо учитывать несколько важных аспектов, включая подходящие численные методы, корректное распределение сетки и постановку условий на границах. В этом ответе мы рассмотрим, как именно можно решить вашу систему ОДЕ, основываясь на представленном коде.

1. Анализ системы ОДЕ

Сначала следует понять структуру ваших уравнений. Убедитесь, что уравнения записаны корректно и что вы правильно определили начальные и граничные условия. Убедитесь, что подгруппы уравнений могут быть решены последовательно. При наличии нескольких переменных (в данном случае F и W) важно правильно определить следующие параметры:

  • Граничные условия: У вас есть условия в пределах r, которые нужно правильно применять на границах.
  • Итерационные методы: Вы можете использовать метод релаксации, чтобы на каждой итерации улучшать значения переменных.

2. Применение метода тридиагональной матрицы (TDMA)

Ваш код использует метод TDMA для решения тридиагональной системы. Это подход подходит для линейных систем, но для нелинейных ОДЕ необходимо применить итерационный процесс, который можно дополнительно обогатить методом подбора.

3. Структура кода

Ваш код выглядит довольно хорошо, но важно убедиться, что все методы и параметры настроены правильно. Обратите внимание на следующие моменты:

  • Инициализация массивов: Убедитесь, что массивы для ваших переменных и коэффициентов инициализированы правильно. Например:

    F = np.ones(N + 2) * 1e-5
    W = np.ones(N + 2) * 1e-5

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

  • Корректировка коэффициентов для границ: Вы установили значения awFL, aeFR, awWL, и aeWR, которые могут вступать в конфликт с другими значениями на границах. Убедитесь, что они правильно подобраны.

  • Метод релаксации: Подбор значения alpha (коэффициент релаксации) может оказать слишком большое влияние на сходимость. Меньшие значения могут замедлить сходимость, в то время как большие могут привести к нестабильности.

4. Рекомендации по улучшению кода

  1. Проверка на NaN и Infinity: Проверьте, происходит ли деление на ноль в коде и добавьте обработку условий NumPy, чтобы избежать появления NaN и Infinity.

  2. Использование scipy.integrate.solve_bvp: Если уравнения позволяют, подумайте о возможностях использования методов из библиотеки SciPy, таких как solve_bvp, которые могут предложить более простой интерфейс для интеграции границ и решения систем ОДЕ.

  3. Визуализация: После завершения расчетов убедитесь, что вы корректно отображаете результаты, используя Matplotlib. Добавьте аннотации для лучшего понимания графиков.

Пример кода

Вот несколько исправлений и улучшений в вашем коде:

import numpy as np
import matplotlib.pyplot as plt

# Функция TDMA для решения уравнений с тридиагональной матрицей
def tdma(...)
    # ваш код

# Параметры
N = 1000
rmin, rmax = 0.0, 100.0
dr = rmax / N
lamda = 10
g = 1.0

# Граничные условия
FL, FR = 0, 1
WL, WR = 0, 1 / rmax

# ...

# Итерационный процесс с улучшением инициализации и проверкой условий
for _ in range(2000):
    # ...

    # Условия для предотвращения перегрузки
    if np.any(np.isnan(F)) or np.any(np.isnan(W)):
        print("Найдено NaN в расчетах")
        break

print(niter, "итераций")

# График результатов
plt.plot(r, F, label="F(r)", color="g")
plt.plot(r, W, label="W(r)", color="r")
plt.legend()
plt.show()

Заключение

Итак, для успешного решения вашей системы нелинейных ОДЕ в Python следует уделить внимание условиям на границах, корректной инициализации переменных, а также тщательной проверке математической валидности вашего кода. Также не забудьте, что использование современных библиотек, таких как SciPy, может значительно упростить процесс решения вам подобных задач.

Оцените материал
Добавить комментарий

Капча загружается...