Merge pull request matplotlib#3771 from thisch/pep8fix_e225_e227_e228

PEP8 : [examples] fix pep8 error classes e225, e227 and e228

Author: tacaswell

examples/event_handling/data_browser.py

@@ -19,7 +19,7 @@ def __init__(self):
     def onpress(self, event):
         if self.lastind is None: return
         if event.key not in ('n', 'p'): return
-        if event.key=='n': inc = 1
+        if event.key == 'n': inc = 1
         else: inc = -1
 
         self.lastind += inc
@@ -28,7 +28,7 @@ def onpress(self, event):
 
     def onpick(self, event):
 
-        if event.artist!=line: return True
+        if event.artist != line: return True
 
         N = len(event.ind)
         if not N: return True
@@ -52,13 +52,13 @@ def update(self):
         ax2.cla()
         ax2.plot(X[dataind])
 
-        ax2.text(0.05, 0.9, 'mu=%1.3f\nsigma=%1.3f'%(xs[dataind], ys[dataind]),
+        ax2.text(0.05, 0.9, 'mu=%1.3f\nsigma=%1.3f' % (xs[dataind], ys[dataind]),
                  transform=ax2.transAxes, va='top')
         ax2.set_ylim(-0.5, 1.5)
         self.selected.set_visible(True)
         self.selected.set_data(xs[dataind], ys[dataind])
 
-        self.text.set_text('selected: %d'%dataind)
+        self.text.set_text('selected: %d' % dataind)
         fig.canvas.draw()
 
 

examples/event_handling/idle_and_timeout.py

@@ -19,12 +19,12 @@
 
 
 def on_idle(event):
-    on_idle.count +=1
+    on_idle.count += 1
     print('idle', on_idle.count)
     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:
+    if on_idle.count == N:
         return False
     return True
 on_idle.cid = None

examples/event_handling/keypress_demo.py

@@ -12,7 +12,7 @@
 def press(event):
     print('press', event.key)
     sys.stdout.flush()
-    if event.key=='x':
+    if event.key == 'x':
         visible = xl.get_visible()
         xl.set_visible(not visible)
         fig.canvas.draw()

examples/event_handling/looking_glass.py

@@ -21,7 +21,7 @@ def __init__(self):
         self.pressevent = None
 
     def onpress(self, event):
-        if event.inaxes!=ax:
+        if event.inaxes != ax:
             return
 
         if not circ.contains(event)[0]:
@@ -34,7 +34,7 @@ def onrelease(self, event):
         self.x0, self.y0 = circ.center
 
     def onmove(self, event):
-        if self.pressevent is None or event.inaxes!=self.pressevent.inaxes:
+        if self.pressevent is None or event.inaxes != self.pressevent.inaxes:
             return
 
         dx = event.xdata - self.pressevent.xdata

examples/event_handling/path_editor.py

@@ -83,15 +83,15 @@ def get_ind_under_point(self, event):
         d = np.sqrt((xt-event.x)**2 + (yt-event.y)**2)
         ind = d.argmin()
 
-        if d[ind]>=self.epsilon:
+        if d[ind] >= self.epsilon:
             ind = None
 
         return ind
 
     def button_press_callback(self, event):
         'whenever a mouse button is pressed'
         if not self.showverts: return
-        if event.inaxes==None: return
+        if event.inaxes == None: return
         if event.button != 1: return
         self._ind = self.get_ind_under_point(event)
 
@@ -104,7 +104,7 @@ def button_release_callback(self, event):
     def key_press_callback(self, event):
         'whenever a key is pressed'
         if not event.inaxes: return
-        if event.key=='t':
+        if event.key == 't':
             self.showverts = not self.showverts
             self.line.set_visible(self.showverts)
             if not self.showverts: self._ind = None

examples/event_handling/pick_event_demo2.py

@@ -18,7 +18,7 @@
 
 def onpick(event):
 
-    if event.artist!=line: return True
+    if event.artist != line: return True
 
     N = len(event.ind)
     if not N: return True
@@ -27,7 +27,7 @@ def onpick(event):
     for subplotnum, dataind in enumerate(event.ind):
         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]),
+        ax.text(0.05, 0.9, 'mu=%1.3f\nsigma=%1.3f' % (xs[dataind], ys[dataind]),
                 transform=ax.transAxes, va='top')
         ax.set_ylim(-0.5, 1.5)
     figi.show()

examples/event_handling/pipong.py

@@ -38,10 +38,10 @@ def __init__(self, disp, x, y, type='l'):
         self.score = 0
         self.xoffset = 0.3
         self.yoffset = 0.1
-        if type=='r':
+        if type == 'r':
             self.xoffset *= -1.0
 
-        if type=='l' or type=='r':
+        if type == 'l' or type == 'r':
             self.signx = -1.0
             self.signy = 1.0
         else:
@@ -54,7 +54,7 @@ def contains(self, loc):
 
 class Puck(object):
     def __init__(self, disp, pad, field):
-        self.vmax= .2
+        self.vmax = .2
         self.disp = disp
         self.field = field
         self._reset(pad)
@@ -76,8 +76,8 @@ def update(self, pads):
         self.y += self.vy
         for pad in pads:
             if pad.contains(self):
-                self.vx *= 1.2 *pad.signx
-                self.vy *= 1.2 *pad.signy
+                self.vx *= 1.2 * pad.signx
+                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:
@@ -121,9 +121,9 @@ class Game(object):
     def __init__(self, ax):
         # create the initial line
         self.ax = ax
-        padAx = padBx= .50
-        padAy = padBy= .30
-        padBx+=6.3
+        padAx = padBx = .50
+        padAy = padBy = .30
+        padBx += 6.3
         pA, = self.ax.barh(padAy, .2, height=.3, color='k', alpha=.5, edgecolor='b', lw=2, label="Player B", animated=True)
         pB, = self.ax.barh(padBy, .2, height=.3, left=padBx, color='k', alpha=.5, edgecolor='r', lw=2, label="Player A", animated=True)
 
@@ -147,7 +147,7 @@ def __init__(self, ax):
         self.pads = []
         self.pads.append(Pad(pA, 0, padAy))
         self.pads.append(Pad(pB, padBx, padBy, 'r'))
-        self.pucks =[]
+        self.pucks = []
         self.i = self.ax.annotate(instructions, (.5, 0.5),
                      name='monospace',
                      verticalalignment='center',
@@ -190,8 +190,8 @@ def draw(self, evt):
             for puck in self.pucks:
                 if puck.update(self.pads):
                     # we only get here if someone scored
-                    self.pads[0].disp.set_label("   "+ str(self.pads[0].score))
-                    self.pads[1].disp.set_label("   "+ str(self.pads[1].score))
+                    self.pads[0].disp.set_label("   " + str(self.pads[0].score))
+                    self.pads[1].disp.set_label("   " + str(self.pads[1].score))
                     self.ax.legend(loc='center')
                     self.leg = self.ax.get_legend()
                     #self.leg.draw_frame(False) #don't draw the legend border
@@ -207,7 +207,7 @@ def draw(self, evt):
         # just redraw the axes rectangle
         self.canvas.blit(self.ax.bbox)
 
-        if self.cnt==50000:
+        if self.cnt == 50000:
             # just so we don't get carried away
             print("...and you've been playing for too long!!!")
             plt.close()

examples/event_handling/poly_editor.py

@@ -74,15 +74,15 @@ def get_ind_under_point(self, event):
         indseq = np.nonzero(np.equal(d, np.amin(d)))[0]
         ind = indseq[0]
 
-        if d[ind]>=self.epsilon:
+        if d[ind] >= self.epsilon:
             ind = None
 
         return ind
 
     def button_press_callback(self, event):
         'whenever a mouse button is pressed'
         if not self.showverts: return
-        if event.inaxes==None: return
+        if event.inaxes == None: return
         if event.button != 1: return
         self._ind = self.get_ind_under_point(event)
 
@@ -95,23 +95,23 @@ def button_release_callback(self, event):
     def key_press_callback(self, event):
         'whenever a key is pressed'
         if not event.inaxes: return
-        if event.key=='t':
+        if event.key == 't':
             self.showverts = not self.showverts
             self.line.set_visible(self.showverts)
             if not self.showverts: self._ind = None
-        elif event.key=='d':
+        elif event.key == 'd':
             ind = self.get_ind_under_point(event)
             if ind is not None:
-                self.poly.xy = [tup for i, tup in enumerate(self.poly.xy) if i!=ind]
+                self.poly.xy = [tup for i, tup in enumerate(self.poly.xy) if i != ind]
                 self.line.set_data(zip(*self.poly.xy))
-        elif event.key=='i':
+        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):
                 s0 = xys[i]
                 s1 = xys[i+1]
                 d = dist_point_to_segment(p, s0, s1)
-                if d<=self.epsilon:
+                if d <= self.epsilon:
                     self.poly.xy = np.array(
                         list(self.poly.xy[:i]) +
                         [(event.xdata, event.ydata)] +

examples/event_handling/zoom_window.py

@@ -28,7 +28,7 @@
 
 
 def onpress(event):
-    if event.button!=1: return
+    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)

examples/misc/image_thumbnail.py

@@ -10,12 +10,12 @@
 import matplotlib.image as image
 
 
-if len(sys.argv)!=2:
-    print('Usage: python %s IMAGEDIR'%__file__)
+if len(sys.argv) != 2:
+    print('Usage: python %s IMAGEDIR' % __file__)
     raise SystemExit
 indir = sys.argv[1]
 if not os.path.isdir(indir):
-    print('Could not find input directory "%s"'%indir)
+    print('Could not find input directory "%s"' % indir)
     raise SystemExit
 
 outdir = 'thumbs'
@@ -26,4 +26,4 @@
     basedir, basename = os.path.split(fname)
     outfile = os.path.join(outdir, basename)
     fig = image.thumbnail(fname, outfile, scale=0.15)
-    print('saved thumbnail of %s to %s'%(fname, outfile))
+    print('saved thumbnail of %s to %s' % (fname, outfile))