from itertools import count
import pandas as pd
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation
import numpy as np
import random

#plt.style.use('fivethirtyeight')

x_vals = []
y_vals = []

index = count()

fig, ax = plt.subplots()
ax.set_xlim(0,100)
ax.set_ylim(0,6)
line, = ax.plot(0, 0)
ax.set_xlabel('x label',fontsize=14)
ax.set_ylabel('y label',fontsize=14)

def animate(i):
    x_vals.append(next(index))
    y_vals.append(random.randint(0,5))

    line.set_xdata(x_vals)
    line.set_ydata(y_vals)
    return line,
    
    #ax.cla()
    #ax.plot(x_vals, y_vals)


#ani = FuncAnimation(plt.gcf(), animate, interval=100)
ani = FuncAnimation(fig, func=animate, frames=np.arange(0,10,1), interval=100)


plt.tight_layout()
plt.show()
