FIX: code convention

rougier · rougier · commit c0dffe1b5b57 · 2014-05-07T19:59:53.000+02:00
diff --git a/examples/animation/rain.py b/examples/animation/rain.py
@@ -4,15 +4,13 @@
 # -----------------------------------------------------------------------------
 import matplotlib
 import numpy as np
-matplotlib.use('TkAgg')
-matplotlib.rcParams['toolbar'] = 'None'
+matplotlib.use('TkAgg') # Required on OSX for animation
 import matplotlib.pyplot as plt
 from  matplotlib.animation import FuncAnimation
 
 
 # Create new figure
 fig = plt.figure(figsize=(7,7))
-fig.patch.set_facecolor('white')
 ax = fig.add_axes([0,0,1,1], frameon=False, aspect=1)
 
 # Create rain data
@@ -23,7 +21,7 @@
 
 # Scatter plot is used to animate rain drops
 scat = ax.scatter(P['position'][:,0], P['position'][:,1], P['size'], lw=0.5,
-                  animated=True, edgecolors = P['color'], facecolors='None')
+                  animated=True, edgecolors = P['color'], facecolors='none')
 ax.set_xlim(0,1), ax.set_xticks([])
 ax.set_ylim(0,1), ax.set_yticks([])
 
@@ -32,22 +30,23 @@ def update(frame):
     i = frame % len(P)
 
     # Make all colors more transparent
-    P['color'][:,3] = np.maximum(0, P['color'][:,3] - 1.0/len(P))
+    P['color'][:,3] -= 1.0/len(P)
+    P['color'][:,3] = np.clip(P['color'][:,3], 0, 1)
 
     # Make all circles bigger
     P['size'] += P['growth']
 
     # Pick a new position for oldest rain drop
-    P['position'][i] = np.random.uniform(0,1,2)
+    P['position'][i] = np.random.uniform(0, 1, 2)
 
     # Reset size
-    P['size'][i]     = 5
+    P['size'][i] = 5
 
     # Reset color
-    P['color'][i]    = 0,0,0,1
+    P['color'][i] = (0, 0, 0, 1)
 
     # Choose a random growth factor
-    P['growth'][i]   = np.random.uniform(50,200)
+    P['growth'][i] = np.random.uniform(50, 200)
 
     # Update scatter plot
     scat.set_edgecolors(P['color'])
