#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
"""