Merge pull request matplotlib#3182 from thisch/pep8examplesapiv2

pep8ify more examples in examples/ + use np.radians/np.degrees

Author: pelson

examples/animation/animate_decay.py

@@ -2,13 +2,14 @@
 import matplotlib.pyplot as plt
 import matplotlib.animation as animation
 
+
 def data_gen():
     t = data_gen.t
     cnt = 0
     while cnt < 1000:
-        cnt+=1
+        cnt += 1
         t += 0.05
-        yield t, np.sin(2*np.pi*t) * np.exp(-t/10.)
+        yield t, np.sin(2 * np.pi * t) * np.exp(-t / 10.)
 data_gen.t = 0
 
 fig, ax = plt.subplots()
@@ -17,20 +18,22 @@ def data_gen():
 ax.set_xlim(0, 5)
 ax.grid()
 xdata, ydata = [], []
+
+
 def run(data):
     # update the data
-    t,y = data
+    t, y = data
     xdata.append(t)
     ydata.append(y)
     xmin, xmax = ax.get_xlim()
 
     if t >= xmax:
-        ax.set_xlim(xmin, 2*xmax)
+        ax.set_xlim(xmin, 2 * xmax)
         ax.figure.canvas.draw()
     line.set_data(xdata, ydata)
 
     return line,
 
 ani = animation.FuncAnimation(fig, run, data_gen, blit=True, interval=10,
-    repeat=False)
+                              repeat=False)
 plt.show()

examples/animation/basic_example.py

@@ -2,8 +2,9 @@
 import matplotlib.pyplot as plt
 import matplotlib.animation as animation
 
+
 def update_line(num, data, line):
-    line.set_data(data[...,:num])
+    line.set_data(data[..., :num])
     return line,
 
 fig1 = plt.figure()
@@ -15,8 +16,8 @@ def update_line(num, data, line):
 plt.xlabel('x')
 plt.title('test')
 line_ani = animation.FuncAnimation(fig1, update_line, 25, fargs=(data, l),
-    interval=50, blit=True)
-#line_ani.save('lines.mp4')
+                                   interval=50, blit=True)
+# line_ani.save('lines.mp4')
 
 fig2 = plt.figure()
 
@@ -28,7 +29,7 @@ def update_line(num, data, line):
     ims.append((plt.pcolor(x, y, base + add, norm=plt.Normalize(0, 30)),))
 
 im_ani = animation.ArtistAnimation(fig2, ims, interval=50, repeat_delay=3000,
-    blit=True)
+                                   blit=True)
 #im_ani.save('im.mp4', metadata={'artist':'Guido'})
 
 plt.show()

examples/animation/basic_example_writer.py

@@ -7,8 +7,9 @@
 import matplotlib.pyplot as plt
 import matplotlib.animation as animation
 
+
 def update_line(num, data, line):
-    line.set_data(data[...,:num])
+    line.set_data(data[..., :num])
     return line,
 
 # Set up formatting for the movie files
@@ -25,7 +26,7 @@ def update_line(num, data, line):
 plt.xlabel('x')
 plt.title('test')
 line_ani = animation.FuncAnimation(fig1, update_line, 25, fargs=(data, l),
-    interval=50, blit=True)
+                                   interval=50, blit=True)
 line_ani.save('lines.mp4', writer=writer)
 
 fig2 = plt.figure()
@@ -38,5 +39,5 @@ def update_line(num, data, line):
     ims.append((plt.pcolor(x, y, base + add, norm=plt.Normalize(0, 30)),))
 
 im_ani = animation.ArtistAnimation(fig2, ims, interval=50, repeat_delay=3000,
-    blit=True)
+                                   blit=True)
 im_ani.save('im.mp4', writer=writer)

examples/animation/bayes_update.py

@@ -4,7 +4,9 @@
 import scipy.stats as ss
 from matplotlib.animation import FuncAnimation
 
+
 class UpdateDist(object):
+
     def __init__(self, ax, prob=0.5):
         self.success = 0
         self.prob = prob
@@ -43,5 +45,5 @@ def __call__(self, i):
 ax = fig.add_subplot(1, 1, 1)
 ud = UpdateDist(ax, prob=0.7)
 anim = FuncAnimation(fig, ud, frames=np.arange(100), init_func=ud.init,
-        interval=100, blit=True)
+                     interval=100, blit=True)
 plt.show()

examples/animation/double_pendulum_animated.py

@@ -7,31 +7,32 @@
 import scipy.integrate as integrate
 import matplotlib.animation as animation
 
-G =  9.8 # acceleration due to gravity, in m/s^2
-L1 = 1.0 # length of pendulum 1 in m
-L2 = 1.0 # length of pendulum 2 in m
-M1 = 1.0 # mass of pendulum 1 in kg
-M2 = 1.0 # mass of pendulum 2 in kg
+G = 9.8  # acceleration due to gravity, in m/s^2
+L1 = 1.0  # length of pendulum 1 in m
+L2 = 1.0  # length of pendulum 2 in m
+M1 = 1.0  # mass of pendulum 1 in kg
+M2 = 1.0  # mass of pendulum 2 in kg
 
 
 def derivs(state, t):
 
     dydx = np.zeros_like(state)
     dydx[0] = state[1]
 
-    del_ = state[2]-state[0]
-    den1 = (M1+M2)*L1 - M2*L1*cos(del_)*cos(del_)
-    dydx[1] = (M2*L1*state[1]*state[1]*sin(del_)*cos(del_)
-               + M2*G*sin(state[2])*cos(del_) + M2*L2*state[3]*state[3]*sin(del_)
-               - (M1+M2)*G*sin(state[0]))/den1
+    del_ = state[2] - state[0]
+    den1 = (M1 + M2) * L1 - M2 * L1 * cos(del_) * cos(del_)
+    dydx[1] = (M2 * L1 * state[1] * state[1] * sin(del_) * cos(del_)
+               + M2 * G * sin(state[2]) * cos(del_) + M2 *
+               L2 * state[3] * state[3] * sin(del_)
+               - (M1 + M2) * G * sin(state[0])) / den1
 
     dydx[2] = state[3]
 
-    den2 = (L2/L1)*den1
-    dydx[3] = (-M2*L2*state[3]*state[3]*sin(del_)*cos(del_)
-               + (M1+M2)*G*sin(state[0])*cos(del_)
-               - (M1+M2)*L1*state[1]*state[1]*sin(del_)
-               - (M1+M2)*G*sin(state[2]))/den2
+    den2 = (L2 / L1) * den1
+    dydx[3] = (-M2 * L2 * state[3] * state[3] * sin(del_) * cos(del_)
+               + (M1 + M2) * G * sin(state[0]) * cos(del_)
+               - (M1 + M2) * L1 * state[1] * state[1] * sin(del_)
+               - (M1 + M2) * G * sin(state[2])) / den2
 
     return dydx
 
@@ -46,19 +47,17 @@ def derivs(state, t):
 th2 = -10.0
 w2 = 0.0
 
-rad = pi/180
-
 # initial state
-state = np.array([th1, w1, th2, w2])*pi/180.
+state = np.radians([th1, w1, th2, w2])
 
 # integrate your ODE using scipy.integrate.
 y = integrate.odeint(derivs, state, t)
 
-x1 = L1*sin(y[:,0])
-y1 = -L1*cos(y[:,0])
+x1 = L1 * sin(y[:, 0])
+y1 = -L1 * cos(y[:, 0])
 
-x2 = L2*sin(y[:,2]) + x1
-y2 = -L2*cos(y[:,2]) + y1
+x2 = L2 * sin(y[:, 2]) + x1
+y2 = -L2 * cos(y[:, 2]) + y1
 
 fig = plt.figure()
 ax = fig.add_subplot(111, autoscale_on=False, xlim=(-2, 2), ylim=(-2, 2))
@@ -68,21 +67,23 @@ def derivs(state, t):
 time_template = 'time = %.1fs'
 time_text = ax.text(0.05, 0.9, '', transform=ax.transAxes)
 
+
 def init():
     line.set_data([], [])
     time_text.set_text('')
     return line, time_text
 
+
 def animate(i):
     thisx = [0, x1[i], x2[i]]
     thisy = [0, y1[i], y2[i]]
 
     line.set_data(thisx, thisy)
-    time_text.set_text(time_template%(i*dt))
+    time_text.set_text(time_template % (i * dt))
     return line, time_text
 
 ani = animation.FuncAnimation(fig, animate, np.arange(1, len(y)),
-    interval=25, blit=True, init_func=init)
+                              interval=25, blit=True, init_func=init)
 
 #ani.save('double_pendulum.mp4', fps=15)
 plt.show()

examples/animation/dynamic_image.py

@@ -8,6 +8,7 @@
 
 fig = plt.figure()
 
+
 def f(x, y):
     return np.sin(x) + np.cos(y)
 
@@ -16,11 +17,12 @@ def f(x, y):
 
 im = plt.imshow(f(x, y), cmap=plt.get_cmap('jet'))
 
+
 def updatefig(*args):
-    global x,y
+    global x, y
     x += np.pi / 15.
     y += np.pi / 20.
-    im.set_array(f(x,y))
+    im.set_array(f(x, y))
     return im,
 
 ani = animation.FuncAnimation(fig, updatefig, interval=50, blit=True)

examples/animation/dynamic_image2.py

@@ -8,6 +8,7 @@
 
 fig = plt.figure()
 
+
 def f(x, y):
     return np.sin(x) + np.cos(y)
 
@@ -24,9 +25,9 @@ def f(x, y):
     ims.append([im])
 
 ani = animation.ArtistAnimation(fig, ims, interval=50, blit=True,
-    repeat_delay=1000)
+                                repeat_delay=1000)
 
-#ani.save('dynamic_images.mp4')
+# ani.save('dynamic_images.mp4')
 
 
 plt.show()

examples/animation/histogram.py

@@ -28,34 +28,36 @@
 # for each rect: 1 for the MOVETO, 3 for the LINETO, 1 for the
 # CLOSEPOLY; the vert for the closepoly is ignored but we still need
 # it to keep the codes aligned with the vertices
-nverts = nrects*(1+3+1)
+nverts = nrects * (1 + 3 + 1)
 verts = np.zeros((nverts, 2))
 codes = np.ones(nverts, int) * path.Path.LINETO
 codes[0::5] = path.Path.MOVETO
 codes[4::5] = path.Path.CLOSEPOLY
-verts[0::5,0] = left
-verts[0::5,1] = bottom
-verts[1::5,0] = left
-verts[1::5,1] = top
-verts[2::5,0] = right
-verts[2::5,1] = top
-verts[3::5,0] = right
-verts[3::5,1] = bottom
+verts[0::5, 0] = left
+verts[0::5, 1] = bottom
+verts[1::5, 0] = left
+verts[1::5, 1] = top
+verts[2::5, 0] = right
+verts[2::5, 1] = top
+verts[3::5, 0] = right
+verts[3::5, 1] = bottom
 
 barpath = path.Path(verts, codes)
-patch = patches.PathPatch(barpath, facecolor='green', edgecolor='yellow', alpha=0.5)
+patch = patches.PathPatch(
+    barpath, facecolor='green', edgecolor='yellow', alpha=0.5)
 ax.add_patch(patch)
 
 ax.set_xlim(left[0], right[-1])
 ax.set_ylim(bottom.min(), top.max())
 
+
 def animate(i):
     # simulate new data coming in
     data = np.random.randn(1000)
     n, bins = np.histogram(data, 100)
     top = bottom + n
-    verts[1::5,1] = top
-    verts[2::5,1] = top
+    verts[1::5, 1] = top
+    verts[2::5, 1] = top
 
 ani = animation.FuncAnimation(fig, animate, 100, repeat=False)
 plt.show()

examples/animation/moviewriter.py

@@ -12,7 +12,7 @@
 
 FFMpegWriter = manimation.writers['ffmpeg']
 metadata = dict(title='Movie Test', artist='Matplotlib',
-        comment='Movie support!')
+                comment='Movie support!')
 writer = FFMpegWriter(fps=15, metadata=metadata)
 
 fig = plt.figure()
@@ -21,12 +21,11 @@
 plt.xlim(-5, 5)
 plt.ylim(-5, 5)
 
-x0,y0 = 0, 0
+x0, y0 = 0, 0
 
 with writer.saving(fig, "writer_test.mp4", 100):
     for i in range(100):
         x0 += 0.1 * np.random.randn()
         y0 += 0.1 * np.random.randn()
         l.set_data(x0, y0)
         writer.grab_frame()
-

examples/animation/rain.py

@@ -8,14 +8,14 @@
 """
 import numpy as np
 import matplotlib.pyplot as plt
-from  matplotlib.animation import FuncAnimation
+from matplotlib.animation import FuncAnimation
 
 
-# Create new Figure and an Axes which fills it. 
-fig = plt.figure(figsize=(7,7))
+# Create new Figure and an Axes which fills it.
+fig = plt.figure(figsize=(7, 7))
 ax = fig.add_axes([0, 0, 1, 1], frameon=False)
-ax.set_xlim(0,1), ax.set_xticks([])
-ax.set_ylim(0,1), ax.set_yticks([])
+ax.set_xlim(0, 1), ax.set_xticks([])
+ax.set_ylim(0, 1), ax.set_yticks([])
 
 # Create rain data
 n_drops = 50
@@ -31,7 +31,7 @@
 
 # Construct the scatter which we will update during animation
 # as the raindrops develop.
-scat = ax.scatter(rain_drops['position'][:,0], rain_drops['position'][:,1],
+scat = ax.scatter(rain_drops['position'][:, 0], rain_drops['position'][:, 1],
                   s=rain_drops['size'], lw=0.5, edgecolors=rain_drops['color'],
                   facecolors='none')
 
@@ -41,8 +41,8 @@ def update(frame_number):
     current_index = frame_number % n_drops
 
     # Make all colors more transparent as time progresses.
-    rain_drops['color'][:, 3] -= 1.0/len(rain_drops)
-    rain_drops['color'][:,3] = np.clip(rain_drops['color'][:,3], 0, 1)
+    rain_drops['color'][:, 3] -= 1.0 / len(rain_drops)
+    rain_drops['color'][:, 3] = np.clip(rain_drops['color'][:, 3], 0, 1)
 
     # Make all circles bigger.
     rain_drops['size'] += rain_drops['growth']
@@ -58,7 +58,7 @@ def update(frame_number):
     scat.set_edgecolors(rain_drops['color'])
     scat.set_sizes(rain_drops['size'])
     scat.set_offsets(rain_drops['position'])
-    
+
 
 # Construct the animation, using the update function as the animation
 # director.