Animation object going down vertically (Z) based on a matplotlib plot / based on a list

Question

I have programmed a simple code of rocket landing just based on its height.

A function is running so that the rocket boosts when it needs to, so it will land safe. A