pep8 fix e2* in examples dir part 1/2

Author: twmr

examples/animation/double_pendulum_animated.py

@@ -19,20 +19,20 @@ 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_)
+    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
+               - (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
+               + (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
 

examples/animation/simple_3danim.py

@@ -22,7 +22,7 @@ def Gen_RandLine(length, dims=2):
         # subtraction by 0.5 is to change the range to [-0.5, 0.5]
         # to allow a line to move backwards.
         step = ((np.random.rand(dims) - 0.5) * 0.1)
-        lineData[:, index] = lineData[:, index-1] + step
+        lineData[:, index] = lineData[:, index - 1] + step
 
     return lineData
 

examples/axes_grid/demo_floating_axes.py

@@ -130,12 +130,12 @@ def setup_axes3(fig, rect):
 
 ax2, aux_ax2 = setup_axes2(fig, 132)
 theta = np.random.rand(10)*.5*np.pi
-radius = np.random.rand(10)+1.
+radius = np.random.rand(10) + 1.
 aux_ax2.scatter(theta, radius)
 
 ax3, aux_ax3 = setup_axes3(fig, 133)
 
-theta = (8 + np.random.rand(10)*(14-8))*15.  # in degrees
+theta = (8 + np.random.rand(10)*(14 - 8))*15.  # in degrees
 radius = np.random.rand(10)*14000.
 aux_ax3.scatter(theta, radius)
 

examples/event_handling/data_browser.py

@@ -23,7 +23,7 @@ def onpress(self, event):
         else: inc = -1
 
         self.lastind += inc
-        self.lastind = np.clip(self.lastind, 0, len(xs)-1)
+        self.lastind = np.clip(self.lastind, 0, len(xs) - 1)
         self.update()
 
     def onpick(self, event):
@@ -37,7 +37,7 @@ def onpick(self, event):
         x = event.mouseevent.xdata
         y = event.mouseevent.ydata
 
-        distances = np.hypot(x-xs[event.ind], y-ys[event.ind])
+        distances = np.hypot(x - xs[event.ind], y - ys[event.ind])
         indmin = distances.argmin()
         dataind = event.ind[indmin]
 

examples/event_handling/idle_and_timeout.py

@@ -21,7 +21,7 @@
 def on_idle(event):
     on_idle.count += 1
     print('idle', on_idle.count)
-    line1.set_ydata(np.sin(2*np.pi*t*(N-on_idle.count)/float(N)))
+    line1.set_ydata(np.sin(2*np.pi*t*(N - on_idle.count)/float(N)))
     event.canvas.draw()
     # test boolean return removal
     if on_idle.count == N:

examples/event_handling/path_editor.py

@@ -80,7 +80,7 @@ def get_ind_under_point(self, event):
         xy = np.asarray(self.pathpatch.get_path().vertices)
         xyt = self.pathpatch.get_transform().transform(xy)
         xt, yt = xyt[:, 0], xyt[:, 1]
-        d = np.sqrt((xt-event.x)**2 + (yt-event.y)**2)
+        d = np.sqrt((xt - event.x)**2 + (yt - event.y)**2)
         ind = d.argmin()
 
         if d[ind] >= self.epsilon:

examples/event_handling/pick_event_demo.py

@@ -120,7 +120,7 @@ def line_picker(line, mouseevent):
         xdata = line.get_xdata()
         ydata = line.get_ydata()
         maxd = 0.05
-        d = np.sqrt((xdata-mouseevent.xdata)**2. + (ydata-mouseevent.ydata)**2.)
+        d = np.sqrt((xdata - mouseevent.xdata)**2. + (ydata - mouseevent.ydata)**2.)
 
         ind = np.nonzero(np.less_equal(d, maxd))
         if len(ind):

examples/event_handling/pick_event_demo2.py

@@ -25,7 +25,7 @@ def onpick(event):
 
     figi = plt.figure()
     for subplotnum, dataind in enumerate(event.ind):
-        ax = figi.add_subplot(N, 1, subplotnum+1)
+        ax = figi.add_subplot(N, 1, subplotnum + 1)
         ax.plot(X[dataind])
         ax.text(0.05, 0.9, 'mu=%1.3f\nsigma=%1.3f' % (xs[dataind], ys[dataind]),
                 transform=ax.transAxes, va='top')

examples/event_handling/pipong.py

@@ -80,17 +80,17 @@ def update(self, pads):
                 self.vy *= 1.2 * pad.signy
         fudge = .001
         # probably cleaner with something like...if not self.field.contains(self.x, self.y):
-        if self.x < 0+fudge:
+        if self.x < 0 + fudge:
             #print("player A loses")
             pads[1].score += 1;
             self._reset(pads[0])
             return True
-        if self.x > 7-fudge:
+        if self.x > 7 - fudge:
             #print("player B loses")
             pads[0].score += 1;
             self._reset(pads[1])
             return True
-        if self.y < -1+fudge or self.y > 1-fudge:
+        if self.y < -1 + fudge or self.y > 1 - fudge:
             self.vy *= -1.0
             # add some randomness, just to make it interesting
             self.vy -= (randn()/300.0 + 1/300.0) * np.sign(self.vy)
@@ -166,10 +166,10 @@ def draw(self, evt):
 
         # show the distractors
         if self.distract:
-            self.line.set_ydata(np.sin(self.x+self.cnt/self.res))
-            self.line2.set_ydata(np.cos(self.x-self.cnt/self.res))
-            self.line3.set_ydata(np.tan(self.x+self.cnt/self.res))
-            self.line4.set_ydata(np.tan(self.x-self.cnt/self.res))
+            self.line.set_ydata(np.sin(self.x + self.cnt/self.res))
+            self.line2.set_ydata(np.cos(self.x - self.cnt/self.res))
+            self.line3.set_ydata(np.tan(self.x + self.cnt/self.res))
+            self.line4.set_ydata(np.tan(self.x - self.cnt/self.res))
             draw_artist(self.line)
             draw_artist(self.line2)
             draw_artist(self.line3)
@@ -224,7 +224,7 @@ def key_press(self, event):
         if event.key == 'e':
             self.pads[0].y += .1
             if self.pads[0].y > 1 - .3:
-                self.pads[0].y = 1-.3
+                self.pads[0].y = 1 - .3
         if event.key == 'd':
             self.pads[0].y -= .1
             if self.pads[0].y < -1:
@@ -233,7 +233,7 @@ def key_press(self, event):
         if event.key == 'i':
             self.pads[1].y += .1
             if self.pads[1].y > 1 - .3:
-                self.pads[1].y = 1-.3
+                self.pads[1].y = 1 - .3
         if event.key == 'k':
             self.pads[1].y -= .1
             if self.pads[1].y < -1:

examples/event_handling/poly_editor.py

@@ -70,7 +70,7 @@ def get_ind_under_point(self, event):
         xy = np.asarray(self.poly.xy)
         xyt = self.poly.get_transform().transform(xy)
         xt, yt = xyt[:, 0], xyt[:, 1]
-        d = np.sqrt((xt-event.x)**2 + (yt-event.y)**2)
+        d = np.sqrt((xt - event.x)**2 + (yt - event.y)**2)
         indseq = np.nonzero(np.equal(d, np.amin(d)))[0]
         ind = indseq[0]
 
@@ -107,9 +107,9 @@ def key_press_callback(self, event):
         elif event.key == 'i':
             xys = self.poly.get_transform().transform(self.poly.xy)
             p = event.x, event.y  # display coords
-            for i in range(len(xys)-1):
+            for i in range(len(xys) - 1):
                 s0 = xys[i]
-                s1 = xys[i+1]
+                s1 = xys[i + 1]
                 d = dist_point_to_segment(p, s0, s1)
                 if d <= self.epsilon:
                     self.poly.xy = np.array(

examples/event_handling/pong_gtk.py

@@ -32,4 +32,4 @@ def start_anim(event):
 tstart = time.time()
 plt.grid()  # to ensure proper background restore
 plt.show()
-print('FPS:' , animation.cnt/(time.time()-tstart))
+print('FPS:', animation.cnt/(time.time() - tstart))

examples/event_handling/zoom_window.py

@@ -30,8 +30,8 @@
 def onpress(event):
     if event.button != 1: return
     x, y = event.xdata, event.ydata
-    axzoom.set_xlim(x-0.1, x+0.1)
-    axzoom.set_ylim(y-0.1, y+0.1)
+    axzoom.set_xlim(x - 0.1, x + 0.1)
+    axzoom.set_ylim(y - 0.1, y + 0.1)
     figzoom.canvas.draw()
 
 figsrc.canvas.mpl_connect('button_press_event', onpress)

examples/misc/ftface_props.py

@@ -17,12 +17,12 @@
 
 font = ft.FT2Font(fname)
 
-print('Num faces   :', font.num_faces)       # number of faces in file
-print('Num glyphs  :', font.num_glyphs)      # number of glyphs in the face
-print('Family name :', font.family_name)     # face family name
-print('Syle name   :', font.style_name)      # face syle name
-print('PS name     :', font.postscript_name) # the postscript name
-print('Num fixed   :', font.num_fixed_sizes) # number of embedded bitmap in face
+print('Num faces   :', font.num_faces)        # number of faces in file
+print('Num glyphs  :', font.num_glyphs)       # number of glyphs in the face
+print('Family name :', font.family_name)      # face family name
+print('Syle name   :', font.style_name)       # face syle name
+print('PS name     :', font.postscript_name)  # the postscript name
+print('Num fixed   :', font.num_fixed_sizes)  # number of embedded bitmap in face
 
 # the following are only available if face.scalable
 if font.scalable:

examples/misc/longshort.py

@@ -34,8 +34,8 @@
 
 
 # long appl, short goog
-g = r.gains1-r.gains2
-tr = (1+g).cumprod()  # the total return
+g = r.gains1 - r.gains2
+tr = (1 + g).cumprod()  # the total return
 
 # plot the return
 fig, ax = plt.subplots()

examples/misc/rc_traits.py

@@ -19,17 +19,17 @@
 flexible_false_trait = traits.Trait(False, flexible_true_trait)
 
 colors = {
-   'c' : '#00bfbf',
-   'b' : '#0000ff',
-   'g' : '#008000',
-   'k' : '#000000',
-   'm' : '#bf00bf',
-   'r' : '#ff0000',
-   'w' : '#ffffff',
-   'y' : '#bfbf00',
-   'gold'                 : '#FFD700',
-   'peachpuff'            : '#FFDAB9',
-   'navajowhite'          : '#FFDEAD',
+   'c': '#00bfbf',
+   'b': '#0000ff',
+   'g': '#008000',
+   'k': '#000000',
+   'm': '#bf00bf',
+   'r': '#ff0000',
+   'w': '#ffffff',
+   'y': '#bfbf00',
+   'gold':        '#FFD700',
+   'peachpuff':   '#FFDAB9',
+   'navajowhite': '#FFDEAD',
    }
 
 

examples/misc/rec_groupby_demo.py

@@ -12,7 +12,7 @@
 def daily_return(prices):
     'an array of daily returns from price array'
     g = np.zeros_like(prices)
-    g[1:] = (prices[1:]-prices[:-1])/prices[:-1]
+    g[1:] = (prices[1:] - prices[:-1])/prices[:-1]
     return g
 
 

examples/misc/svg_filter_line.py

@@ -36,7 +36,7 @@
     shadow.set_color("0.2")
     # adjust zorder of the shadow lines so that it is drawn below the
     # original lines
-    shadow.set_zorder(l.get_zorder()-0.5)
+    shadow.set_zorder(l.get_zorder() - 0.5)
 
     # offset transform
     ot = mtransforms.offset_copy(l.get_transform(), fig1,
@@ -45,7 +45,7 @@
     shadow.set_transform(ot)
 
     # set the id for a later use
-    shadow.set_gid(l.get_label()+"_shadow")
+    shadow.set_gid(l.get_label() + "_shadow")
 
 
 ax.set_xlim(0., 1.)
@@ -77,7 +77,7 @@
 
 for l in [l1, l2]:
     # pick up the svg element with given id
-    shadow = xmlid[l.get_label()+"_shadow"]
+    shadow = xmlid[l.get_label() + "_shadow"]
     # apply shdow filter
     shadow.set("filter", 'url(#dropshadow)')
 

examples/misc/svg_filter_pie.py

@@ -36,7 +36,7 @@
 for w in pies[0]:
     # create shadow patch
     s = Shadow(w, -0.01, -0.01)
-    s.set_gid(w.get_gid()+"_shadow")
+    s.set_gid(w.get_gid() + "_shadow")
     s.set_zorder(w.get_zorder() - 0.1)
     ax.add_patch(s)
 

examples/mplot3d/hist3d_demo.py

@@ -8,7 +8,7 @@
 hist, xedges, yedges = np.histogram2d(x, y, bins=4)
 
 elements = (len(xedges) - 1) * (len(yedges) - 1)
-xpos, ypos = np.meshgrid(xedges[:-1]+0.25, yedges[:-1]+0.25)
+xpos, ypos = np.meshgrid(xedges[:-1] + 0.25, yedges[:-1] + 0.25)
 
 xpos = xpos.flatten()
 ypos = ypos.flatten()

examples/mplot3d/lorenz_attractor.py

@@ -11,7 +11,7 @@
 from mpl_toolkits.mplot3d import Axes3D
 
 
-def lorenz(x, y, z, s=10, r=28, b=2.667) :
+def lorenz(x, y, z, s=10, r=28, b=2.667):
     x_dot = s*(y - x)
     y_dot = r*x - y - x*z
     z_dot = x*y - b*z
@@ -30,7 +30,7 @@ def lorenz(x, y, z, s=10, r=28, b=2.667) :
 xs[0], ys[0], zs[0] = (0., 1., 1.05)
 
 # Stepping through "time".
-for i in range(stepCnt) :
+for i in range(stepCnt):
     # Derivatives of the X, Y, Z state
     x_dot, y_dot, z_dot = lorenz(xs[i], ys[i], zs[i])
     xs[i + 1] = xs[i] + (x_dot * dt)

examples/mplot3d/scatter3d_demo.py

@@ -4,7 +4,7 @@
 
 
 def randrange(n, vmin, vmax):
-    return (vmax-vmin)*np.random.rand(n) + vmin
+    return (vmax - vmin)*np.random.rand(n) + vmin
 
 fig = plt.figure()
 ax = fig.add_subplot(111, projection='3d')

examples/pylab_examples/alignment_test.py

@@ -48,24 +48,24 @@
         verticalalignment='top',
         transform=ax.transAxes)
 
-ax.text(left, 0.5*(bottom+top), 'right center',
+ax.text(left, 0.5*(bottom + top), 'right center',
         horizontalalignment='right',
         verticalalignment='center',
         rotation='vertical',
         transform=ax.transAxes)
 
-ax.text(left, 0.5*(bottom+top), 'left center',
+ax.text(left, 0.5*(bottom + top), 'left center',
         horizontalalignment='left',
         verticalalignment='center',
         rotation='vertical',
         transform=ax.transAxes)
 
-ax.text(0.5*(left+right), 0.5*(bottom+top), 'middle',
+ax.text(0.5*(left + right), 0.5*(bottom + top), 'middle',
         horizontalalignment='center',
         verticalalignment='center',
         transform=ax.transAxes)
 
-ax.text(right, 0.5*(bottom+top), 'centered',
+ax.text(right, 0.5*(bottom + top), 'centered',
         horizontalalignment='center',
         verticalalignment='center',
         rotation='vertical',

examples/pylab_examples/anscombe.py

@@ -20,7 +20,7 @@
 
 
 def fit(x):
-    return 3+0.5*x
+    return 3 + 0.5*x
 
 
 xfit = array([amin(x), amax(x)])