aaa | Python Fiddle

# -*- coding: utf-8 -*-
"""
Created on Tue May 26 23:47:48 2015

@author: gal
"""

import matplotlib.pyplot as plt
import numpy as np
from matplotlib import animation

plt.rc("text", usetex = False)
plt.rc("font", size = 14, family = "serif", serif = ["Arial"])
plt.rc("xtick", labelsize = "small")
plt.rc("ytick", labelsize = "small")
plt.rc("legend", frameon = False, fontsize = "medium")
plt.rc("figure", figsize = (10, 6))
plt.rc("lines", linewidth=2.0)
plt.rc("axes", color_cycle=["k"])

plt.close("all")
#------------------------------------------------------------------------------

fig, axes = plt.subplots(3,sharex = False, facecolor = "white")

m1 = 30 * 10**6
mg1 = (3/4) * m1
m2 = 10 * 10**6
mg2 = (3/4) * m2

pozicija,=axes[0].plot([],[],'r>', markersize = 10)

def init():
    axes[0].set_xlim(0,600000)

def raketa(x):
    t = x
    vr = 3000
    fim = 100
    m1 = 30 * 10**3
    m2 = 10 * 10**3
    M = m1 + m2 - (fim*t)
    if t > 225:
        M -= 0.25 * m1
        pozicija.set_data([vr*(t-225)*np.log((m1+m2)/(m2+m1/4)) + ((vr/fim)*(m1+m2-225*fim)*(np.log((1-fim*225/(m1+m2)))-1) + vr*(m1+m2)/fim) + vr/fim*(m2-fim*(t-225))*(np.log((1-fim*(t-225)/(m2)))-1) + vr*m2/fim],[0])
        axes[2].plot(t,vr * np.log((m1+m2)/(m2+ 0.25*m1)) + vr * np.log(m2/M),'.r')
    else:
        axes[2].plot(t,vr * np.log((m1+m2)/M),'.r')
        pozicija.set_data([(vr/fim)*M*(np.log((1-fim*t/(m1+m2)))-1) + vr*(m1+m2)/fim],[0])
    F = fim * vr
    axes[1].plot(t,F/M,'.r')

animation.FuncAnimation(fig,raketa,frames = 300, init_func=init, interval = 10)

Python Fiddle

Python Cloud IDE