---

import numpy as np

nx, nt = 5, 3
hx, ht = 1/4, 1/2
xl, xn = 0, 1
tl, tn = 0, 1

U = [[1 for j in range(nt)] for i in range(nx)]  # начальное приближение

X = [hx * i for i in range(nx)]
T = [ht * j for j in range(nt)]

fi = [96*t*(1-t) for t in T]
psi = [44*x*(1-x) for x in X]
F = -280

# граничные условия задачи Дирихле
U = np.array(U)

U[0], U[nx - 1] = fi, fi
U[:, 0], U[:, nt - 1] = psi, psi

U