import pandas as pd import numpy as np from scipy import integrate, optimize import matplotlib.pyplot as plt
定义SEIR模型:
# define SEIR model defSEIR_model(Y,t,beta,q,a,sigma,epsilon,gamma): S,E,I,R = Y dS = - beta * S * (1-a) * I / N - beta * S * (1-a) * q * E / N - beta * a * S * (1-sigma) * I / N - beta * a * S * (1-sigma) * q * E / N - sigma * a * S dE = beta * S * (1-a) * I / N + beta * S * (1-a) * q * E / N + beta * a * S * (1-sigma) * I / N + beta * a * S * (1-sigma) * q * E / N - epsilon * E dI = epsilon * E - gamma * I dR = sigma * a * S + gamma * I return dS,dE,dI,dR