import numpy as np

# set values of constants

v_0 = 5     # initial velocity
y_0 = 200   # initial y position
g = 9.81    # acceleration due to gravity

# calculate y for integer values of t from 0 to 9

t = np.arange(0, 10)
y = y_0 - 0.5 * g * t ** 2
x = v_0 * t

print("t (seconds):", t)
print("y (metres):", y)
print("x (metres):", x)

import matplotlib.pyplot as plt

plt.figure(figsize=(5,5))
plt.plot(t, y)

plt.figure(figsize=(5,5))
plt.plot(t, x)
plt.figure(figsize=(5,5))
plt.plot(x, y)

t = np.arange(0, 365)
d = 149600000
p = 365



x = d * np.cos(2 * np.pi * t * (1 / p)) 
y = d * np.sin(2 * np.pi * t * (1 / p)) 

from matplotlib import animation
from IPython.display import HTML, display
import random

filename = "animation.gif"
# Number of animation frames equals the length of time array t
frames = len(t) // 
interval = 100

def ganimate(frame):
    plt.cla()
    plt.scatter(x[frame], y[frame])
    plt.scatter(0, 0)
    plt.xlim(-159600000, 159600000)
    plt.ylim(-159600000, 159600000)

fig = plt.figure(figsize=(5, 5))
anim = animation.FuncAnimation(fig, ganimate, frames=frames, interval=interval)
anim.save(filename, writer='imagemagick')
plt.close()

__counter__ = str(random.randint(0,2e9))
display(HTML('<img src="' + filename + '?' + __counter__ + '">'))