Update Matplotlib frame by frame in jupyter notebook for multiple lines

notebook_cell.py

from time import sleep
%matplotlib notebook


x = []

fig = plt.figure()
ax = fig.add_subplot(1,1,1)
ax.autoscale()
num_agents = 5
y=np.random.randn(1,num_agents)
print(y.shape)
lines = ax.plot([0], y)
print(len(lines))
# help(ax.plot)

for i in range(30):
    y = np.concatenate( (y, np.random.randn(1,num_agents)), axis = 0)
    for li, line in enumerate(lines):
        line.set_xdata(range(y.shape[0]))
        line.set_ydata(y[:,li])
    ax.relim()
    ax.autoscale_view()
    fig.canvas.draw()
    sleep(0.1)