#Exer. 5.10 from "A primer on..."
import numpy as np
import matplotlib.pyplot as plt
import sys
v0_list = sys.argv[1:]
g = 9.81
for v0 in v0_list:
v0 = float(v0)
t_stop = 2*v0/g
t = np.linspace(0,t_stop,101)
y = v0*t - 0.5*g*t**2
plt.plot(t,y)
plt.title('Ball plot 2')
plt.xlabel('time (s)')
plt.ylabel('height (m)')
plt.show()
"""
Terminal> python plot_ball2.py 3 5 9
output is a plot
"""