from pylab import *
from numpy import *  #je dobre zvyknutis si aj na tento import
#uvedene dva importy v neinteraktivne spustanych .py suboroch robia nieco podobne
# ako ked sa interaktivny ipython spustkik s option --pylab 

omega = 15.
b=0.25
A=2.
t=empty(500) # nenaplnene array casovych momentov indexovanych cislami od 0 po 499
x=empty(500) # nenaplnene array vychyliek indexovanych cislami od 0 po 499
y=empty(500) # nenaplnene array hodnot exponencialy indexovanych cislami od 0 po 499
dt=10./500.  # maly krok v case, 10 sekund rozdelena na 500 intervalov

tau = double(input("zadaj dobu zivota v sekundach   "))
print("tau= ",tau) 

for i in range(0,500):    #for statement, i bude z mnoziny {0,1,2,3,...,499}
	t[i]=i*dt  #casovy okamih
	x[i]=A*exp(-b*t[i])*cos(omega*t[i])
	y[i]=A*exp.....  # dopln klesajucu exponencialu s dobou zivota tau
plot(t,x)     # vykresli kmity tlmeneho oscilatora
plot(t,y)      #vykresli tlmenu excpomencialu
show()  #okno s grafom treba zatvorit aby to skoncilo