import matplotlib.pyplot as plt
import numpy as np
# plot x as fn of t
k = 5.0
x0 = 10.0
x = x0
m = 2.0
time = 6.0
displacementlist = []
velocitylist = []
timelist = np.arange(0, 6.0, 0.01)
v = 0
dt = 0.001
t = 0
for i in range(len(timelist)):
    acc = (x * -k) / m
    v = v + (acc * timelist[i])
    x = x + (v * timelist[i])
    displacementlist.append(x)
    velocitylist.append(v)
print(timelist)
print(velocitylist)
print(displacementlist)

plt.plot(timelist, velocitylist, color = 'r')
plt.plot(timelist, displacementlist, color = 'b')
plt.axhline(0)
plt.show()