1535

import numpy as np
import matplotlib.pyplot as plt

Q = 200
P = np.zeros(Q)
P[0] = 1
PP = np.zeros(Q)
t_max = 100
dt = 0.001
lmbd = 4
mu = 5
t = 0
p0 = [1]
p1 = []

while t < t_max:
    PP[0] = dt * (-lmbd * P[0] + mu * P[1])
    for j in range(1, Q-1):
        PP[j] = dt * (-(lmbd + mu) * P[j] + lmbd * P[j-1] + mu * P[j + 1])
    for j in range(Q):
        P[j] += PP[j]
        if j == 0: p0.append()
        t += dt
print(P)
t_to_plot = np.linspace(0, 100, 200)
plt.plot(t_to_plot, P)
plt.plot()
plt.show()