Fixed snippet

Author: CoreyMSchafer

Python/Matplotlib/09-LiveData/snippets.txt

@@ -1,15 +1,30 @@
 # Another way to do it without clearing the Axis
+from itertools import count
+import pandas as pd
+import matplotlib.pyplot as plt
+from matplotlib.animation import FuncAnimation
+
+plt.style.use('fivethirtyeight')
+
+x_vals = []
+y_vals = []
+
+plt.plot([], [], label='Channel 1')
+plt.plot([], [], label='Channel 2')
+
 
 def animate(i):
  data = pd.read_csv('data.csv')
  x = data['x_value']
  y1 = data['total_1']
  y2 = data['total_2']
 
+ ax = plt.gca()
+ line1, line2 = ax.lines
+
  line1.set_data(x, y1)
  line2.set_data(x, y2)
 
- ax = plt.gca()
  xlim_low, xlim_high = ax.get_xlim()
  ylim_low, ylim_high = ax.get_ylim()
 
@@ -24,3 +39,10 @@ def animate(i):
  current_ymin = y1min if (y1min < y2min) else y2min
 
  ax.set_ylim((current_ymin - 5), (current_ymax + 5))
+
+
+ani = FuncAnimation(plt.gcf(), animate, interval=1000)
+
+plt.legend()
+plt.tight_layout()
+plt.show()