Plot reversals

rob-luke · Mar 29, 2015 · e0a3782 · e0a3782
1 parent 003e288
commit e0a3782
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/UpDownMethods/plot.py b/UpDownMethods/plot.py
@@ -24,9 +24,11 @@ def plot_results(results, midpoints=False, figure=None, estimate=False,
         plt.scatter(incorr.index+1, incorr.Value, s=50, marker='_', c='k')
 
     # Indicate reversals
-    # reversal = results[results['Reversal'] == True]
-    # plt.scatter(reversal.index+1, reversal.Value, facecolors='none',
-    # edgecolors='k', s=200)
+    if reversals:
+        reversal = results[results['Reversal'] == True]
+        if len(reversal) > 0:
+            plt.scatter(reversal.index+1, reversal.Value, facecolors='none',
+                        edgecolors='k', s=200)
 
     # Track the runs
     runs = ud.runs(results)