Как составить график, который показывает исходную плотность деревьев по «оси x» и плотность сожженных деревьев по «оси y»?

Question

Я делаю это школьное задание, и мне нужна помощь (ro = плотность).

Вот код:

import forest_fire
import matplotlib.pyplot as plt
def get_trees(ro):
    forest = forest_fire.Forest(ro = ro, n = 100)
    n0 = forest.count
    forest.loop(100)
    return forest.count / n0
ro_list = list(map(lambda x:0.05 * x,range(1,21)))
trees_not_burned = list(map(get_trees,ro_list))
plt.plot(ro_list,trees_not_burned,'r.-')
plt.show()