In [5]:

    

import math 
import numpy as np
t_max = 10 # maximum time
h = 0.1 # time step
g = -9.81
n_steps = int(t_max/h)
t = np.zeros(n_steps)
v = np.zeros(n_steps)
 
# set initial conditions
t[0] = 0
v[0] = 0
 
for i in range(1,n_steps):
    t[i] = t[i-1] + h
    v[i] = v[i-1] + h * g
print t
print v


    

    




[ 0.   0.1  0.2  0.3  0.4  0.5  0.6  0.7  0.8  0.9  1.   1.1  1.2  1.3  1.4
  1.5  1.6  1.7  1.8  1.9  2.   2.1  2.2  2.3  2.4  2.5  2.6  2.7  2.8  2.9
  3.   3.1  3.2  3.3  3.4  3.5  3.6  3.7  3.8  3.9  4.   4.1  4.2  4.3  4.4
  4.5  4.6  4.7  4.8  4.9  5.   5.1  5.2  5.3  5.4  5.5  5.6  5.7  5.8  5.9
  6.   6.1  6.2  6.3  6.4  6.5  6.6  6.7  6.8  6.9  7.   7.1  7.2  7.3  7.4
  7.5  7.6  7.7  7.8  7.9  8.   8.1  8.2  8.3  8.4  8.5  8.6  8.7  8.8  8.9
  9.   9.1  9.2  9.3  9.4  9.5  9.6  9.7  9.8  9.9]
[  0.     -0.981  -1.962  -2.943  -3.924  -4.905  -5.886  -6.867  -7.848
  -8.829  -9.81  -10.791 -11.772 -12.753 -13.734 -14.715 -15.696 -16.677
 -17.658 -18.639 -19.62  -20.601 -21.582 -22.563 -23.544 -24.525 -25.506
 -26.487 -27.468 -28.449 -29.43  -30.411 -31.392 -32.373 -33.354 -34.335
 -35.316 -36.297 -37.278 -38.259 -39.24  -40.221 -41.202 -42.183 -43.164
 -44.145 -45.126 -46.107 -47.088 -48.069 -49.05  -50.031 -51.012 -51.993
 -52.974 -53.955 -54.936 -55.917 -56.898 -57.879 -58.86  -59.841 -60.822
 -61.803 -62.784 -63.765 -64.746 -65.727 -66.708 -67.689 -68.67  -69.651
 -70.632 -71.613 -72.594 -73.575 -74.556 -75.537 -76.518 -77.499 -78.48
 -79.461 -80.442 -81.423 -82.404 -83.385 -84.366 -85.347 -86.328 -87.309
 -88.29  -89.271 -90.252 -91.233 -92.214 -93.195 -94.176 -95.157 -96.138
 -97.119]

In [ ]: