Is there a better way to write this in python? jupyter notebook

Question

You will be responsible for writing a program that simulates the spread of an infectious disease throughout a population similar to the previous program. Using classes, you