fix formatting

- fix E12,E231,E241,E251,E26,E30 with autopep8
- fix the rest by hand

Author: jimustafa

docs/conf.py

@@ -5,8 +5,8 @@
 html_title = 'Visualization with Python'
 project = "Matplotlib cheatsheets"
 copyright = (
-     f"2012 - {datetime.datetime.now().year} The Matplotlib development team"
- )
+    f"2012 - {datetime.datetime.now().year} The Matplotlib development team"
+)
 author = "Matplotlib Developers"
 
 # -- General configuration ---------------------------------------------------

logos/mpl-logos2.py

@@ -155,6 +155,7 @@ def make_logo(height_px, lw_bars, lw_grid, lw_border, rgrid, with_text=False):
 
     return fig, ax
 
+
 ##############################################################################
 # A large logo:
 

scripts/adjustements.py

@@ -10,39 +10,39 @@
 
 
 fig = plt.figure(figsize=(4.25, 4.25 * 95/115))
-ax = fig.add_axes([0,0,1,1], frameon=False, aspect=1,
-                  xlim=(0-5,100+10), ylim=(-10,80+5), xticks=[], yticks=[])
+ax = fig.add_axes([0, 0, 1, 1], frameon=False, aspect=1,
+                  xlim=(0-5, 100+10), ylim=(-10, 80+5), xticks=[], yticks=[])
 
 
 box = mpatches.FancyBboxPatch(
-    (0,0), 100, 83,  mpatches.BoxStyle("Round", pad=0, rounding_size=2),
+    (0, 0), 100, 83, mpatches.BoxStyle("Round", pad=0, rounding_size=2),
     linewidth=1., facecolor="0.9", edgecolor="black")
 ax.add_artist(box)
 
 box = mpatches.FancyBboxPatch(
-    (0,0), 100, 75, mpatches.BoxStyle("Round", pad=0, rounding_size=0),
+    (0, 0), 100, 75, mpatches.BoxStyle("Round", pad=0, rounding_size=0),
     linewidth=1., facecolor="white", edgecolor="black")
 ax.add_artist(box)
 
 
 box = mpatches.Rectangle(
-    (5,5), 45, 30, zorder=10,
+    (5, 5), 45, 30, zorder=10,
     linewidth=1.0, facecolor="white", edgecolor="black")
 ax.add_artist(box)
 
 box = mpatches.Rectangle(
-    (5,40), 45, 30, zorder=10,
+    (5, 40), 45, 30, zorder=10,
     linewidth=1.0, facecolor="white", edgecolor="black")
 ax.add_artist(box)
 
 box = mpatches.Rectangle(
-    (55,5), 40, 65, zorder=10,
+    (55, 5), 40, 65, zorder=10,
     linewidth=1.0, facecolor="white", edgecolor="black")
 ax.add_artist(box)
 
 # Window button
-X, Y = [5,10,15], [79,79,79]
-plt.scatter(X, Y, s=75,  zorder=10,
+X, Y = [5, 10, 15], [79, 79, 79]
+plt.scatter(X, Y, s=75, zorder=10,
             edgecolor="black", facecolor="white", linewidth=1)
 
 
@@ -60,7 +60,7 @@
 plt.plot(X, Y, color="black", linestyle=":", linewidth=1, clip_on=False)
 
 
-def ext_arrow(p0,p1,p2,p3):
+def ext_arrow(p0, p1, p2, p3):
     p0, p1 = np.asarray(p0), np.asarray(p1)
     p2, p3 = np.asarray(p2), np.asarray(p3)
     ax.arrow(*p0, *(p1-p0), zorder=20, linewidth=0,
@@ -69,9 +69,10 @@ def ext_arrow(p0,p1,p2,p3):
     ax.arrow(*p3, *(p2-p3), zorder=20, linewidth=0,
              length_includes_head=True, width=.4,
              head_width=2, head_length=2, color="black")
-    plt.plot([p1[0],p2[0]], [p1[1],p2[1]], linewidth=.9, color="black")
+    plt.plot([p1[0], p2[0]], [p1[1], p2[1]], linewidth=.9, color="black")
 
-def int_arrow(p0,p1):
+
+def int_arrow(p0, p1):
     p0, p1 = np.asarray(p0), np.asarray(p1)
     ax.arrow(*((p0+p1)/2), *((p1-p0)/2), zorder=20, linewidth=0,
              length_includes_head=True, width=.4,
@@ -81,48 +82,47 @@ def int_arrow(p0,p1):
              head_width=2, head_length=2, color="black")
 
 
-
 x = 0
 y = 10
-ext_arrow( (x-4,y), (x,y), (x+5,y), (x+9,y) )
+ext_arrow( (x-4, y), (x, y), (x+5, y), (x+9, y) )
 ax.text(x+9.5, y, "left", ha="left", va="center", size="x-small", zorder=20)
 
 x += 50
-ext_arrow( (x-4,y), (x,y), (x+5,y), (x+9,y) )
+ext_arrow( (x-4, y), (x, y), (x+5, y), (x+9, y) )
 ax.text(x-4.5, y, "wspace", ha="right", va="center", size="x-small", zorder=20)
 
 x += 45
-ext_arrow( (x-4,y), (x,y), (x+5,y), (x+9,y) )
+ext_arrow( (x-4, y), (x, y), (x+5, y), (x+9, y) )
 ax.text(x-4.5, y, "right", ha="right", va="center", size="x-small", zorder=20)
 
 y = 0
 x = 25
-ext_arrow( (x,y-4), (x,y), (x,y+5), (x,y+9) )
+ext_arrow( (x, y-4), (x, y), (x, y+5), (x, y+9) )
 ax.text(x, y+9.5, "bottom", ha="center", va="bottom", size="x-small", zorder=20)
 
 y += 35
-ext_arrow( (x,y-4), (x,y), (x,y+5), (x,y+9) )
+ext_arrow( (x, y-4), (x, y), (x, y+5), (x, y+9) )
 ax.text(x, y-4.5, "hspace", ha="center", va="top", size="x-small", zorder=20)
 
 y += 35
-ext_arrow( (x,y-4), (x,y), (x,y+5), (x,y+9) )
+ext_arrow( (x, y-4), (x, y), (x, y+5), (x, y+9) )
 ax.text(x, y-4.5, "top", ha="center", va="top", size="x-small", zorder=20)
 
-int_arrow((0,-5), (100,-5))
+int_arrow((0, -5), (100, -5))
 ax.text(50, -5, "figure width", backgroundcolor="white", zorder=30,
         ha="center", va="center", size="x-small")
 
-int_arrow((105,0), (105,75))
+int_arrow((105, 0), (105, 75))
 ax.text(105, 75/2, "figure height", backgroundcolor="white", zorder=30,
-        rotation = "vertical", ha="center", va="center", size="x-small")
+        rotation="vertical", ha="center", va="center", size="x-small")
 
-int_arrow((55,62.5), (95,62.5))
+int_arrow((55, 62.5), (95, 62.5))
 ax.text(75, 62.5, "axes width", backgroundcolor="white", zorder=30,
         ha="center", va="center", size="x-small")
 
-int_arrow((62.5,5), (62.5,70))
+int_arrow((62.5, 5), (62.5, 70))
 ax.text(62.5, 35, "axes height", backgroundcolor="white", zorder=30,
-        rotation = "vertical", ha="center", va="center", size="x-small")
+        rotation="vertical", ha="center", va="center", size="x-small")
 
 
 plt.savefig("../figures/adjustments.pdf")

scripts/anatomy.py

@@ -23,6 +23,7 @@ def minor_tick(x, pos):
         return ""
     return "%.2f" % x
 
+
 ax.xaxis.set_major_locator(MultipleLocator(1.000))
 ax.xaxis.set_minor_locator(AutoMinorLocator(4))
 ax.yaxis.set_major_locator(MultipleLocator(1.000))
@@ -124,13 +125,13 @@ def text(x, y, text):
 
 color = '#000099'
 ax.annotate('Spines', xy=(4.0, 0.35), xytext=(3.3, 0.5), color=color,
-            weight='regular', # fontsize="large", fontname="Yanone Kaffeesatz",
+            weight='regular',  # fontsize="large", fontname="Yanone Kaffeesatz",
             arrowprops=dict(arrowstyle='->',
                             connectionstyle="arc3",
                             color=color))
 
 ax.annotate('', xy=(3.15, 0.0), xytext=(3.45, 0.45), color=color,
-            weight='regular', # fontsize="large", fontname="Yanone Kaffeesatz",
+            weight='regular',  # fontsize="large", fontname="Yanone Kaffeesatz",
             arrowprops=dict(arrowstyle='->',
                             connectionstyle="arc3",
                             color=color))

scripts/animation.py

@@ -2,11 +2,15 @@
 import matplotlib.pyplot as plt
 import matplotlib.animation as animation
 
-T = np.linspace(0,2*np.pi,100)
+T = np.linspace(0, 2*np.pi, 100)
 S = np.sin(T)
 line, = plt.plot(T, S)
+
+
 def animate(i):
     line.set_ydata(np.sin(T+i/50))
+
+
 a=animation.FuncAnimation(
     plt.gcf(), animate, interval=5)
 # plt.show()

scripts/annotate.py

@@ -7,18 +7,17 @@
 import matplotlib.pyplot as plt
 
 
-fig = plt.figure(figsize=(6,1))
-#ax = plt.subplot(111, frameon=False, aspect=.1)
+fig = plt.figure(figsize=(6, 1))
+# ax = plt.subplot(111, frameon=False, aspect=.1)
 # b = 0.0
-ax = fig.add_axes([0,0,1,1], frameon=False, aspect=1)
+ax = fig.add_axes([0, 0, 1, 1], frameon=False, aspect=1)
 
 
-
-plt.scatter([5.5],[0.75], s=100, c="k")
-plt.xlim(0,6), plt.ylim(0,1)
+plt.scatter([5.5], [0.75], s=100, c="k")
+plt.xlim(0, 6), plt.ylim(0, 1)
 plt.xticks([]), plt.yticks([])
 
-plt.annotate("Annotation", (5.5,.75), (0.1,.75), size=16, va="center",
+plt.annotate("Annotation", (5.5, .75), (0.1, .75), size=16, va="center",
              arrowprops=dict(facecolor='black', shrink=0.05))
 
 plt.text( 5.5, 0.6, "xy\nycoords", size=10, va="top", ha="center", color=".5")

scripts/annotation-arrow-styles.py

@@ -2,15 +2,18 @@
 import matplotlib.patches as mpatches
 
 styles = mpatches.ArrowStyle.get_styles()
+
+
 def demo_con_style(ax, connectionstyle):
     ax.text(.05, .95, connectionstyle.replace(",", ",\n"),
             family="Source Code Pro",
             transform=ax.transAxes, ha="left", va="top", size="x-small")
 
+
 (fig, axes) = plt.subplots(5, 3, figsize=(4, 2.5), frameon=False)
 for ax in axes.flatten():
     ax.axis("off")
-for i,(ax,style) in enumerate(zip(axes.flatten(), mpatches.ArrowStyle.get_styles())):
+for i, (ax, style) in enumerate(zip(axes.flatten(), mpatches.ArrowStyle.get_styles())):
     x0, y0 = 0.8, 0.5
     x1, y1 = 0.2, 0.5
     ax.plot([x0, x1], [y0, y1], ".", color="0.25")
@@ -24,7 +27,7 @@ def demo_con_style(ax, connectionstyle):
                                 connectionstyle="arc3,rad=0"))
     ax.text( (x1+x0)/2, y0-0.2, style,
              transform=ax.transAxes,
-             family = "Source Code Pro", ha="center", va="top")
+             family="Source Code Pro", ha="center", va="top")
 
 plt.savefig("../figures/annotation-arrow-styles.pdf")
 # plt.show()

scripts/annotation-connection-styles.py

@@ -1,5 +1,6 @@
 import matplotlib.pyplot as plt
 
+
 def demo_con_style(ax, connectionstyle):
     x1, y1 = 0.3, 0.2
     x2, y2 = 0.8, 0.6
@@ -16,6 +17,7 @@ def demo_con_style(ax, connectionstyle):
             family="Source Code Pro",
             transform=ax.transAxes, ha="left", va="top", size="x-small")
 
+
 fig, axs = plt.subplots(3, 3, figsize=(5, 5))
 demo_con_style(axs[0, 0], "arc3,rad=0")
 demo_con_style(axs[0, 1], "arc3,rad=0.3")

scripts/basic-plots.py

@@ -68,10 +68,10 @@
 # Basic imshow plot
 # -----------------------------------------------------------------------------
 np.random.seed(3)
-I = np.zeros((8, 8, 4))
-I[:,:] = mpl.colors.to_rgba("C1")
-I[...,3] = np.random.uniform(0.25, 1.0, (8, 8))
-ax.imshow(I, extent=[0, 8, 0, 8], interpolation="nearest")
+Z = np.zeros((8, 8, 4))
+Z[:, :] = mpl.colors.to_rgba("C1")
+Z[..., 3] = np.random.uniform(0.25, 1.0, (8, 8))
+ax.imshow(Z, extent=[0, 8, 0, 8], interpolation="nearest")
 ax.set_xlim(0, 8), ax.set_xticks(np.arange(1, 8))
 ax.set_ylim(0, 8), ax.set_yticks(np.arange(1, 8))
 ax.grid(linewidth=0.25, color="white")

scripts/colorbar.py

@@ -7,16 +7,16 @@
 import matplotlib.pyplot as plt
 
 
-fig = plt.figure(figsize=(6,.65))
-#ax = plt.subplot(111, frameon=False, aspect=.1)
+fig = plt.figure(figsize=(6, .65))
+# ax = plt.subplot(111, frameon=False, aspect=.1)
 b = 0.025
-ax = fig.add_axes([b,10*b,1-2*b,1-10*b], frameon=False, aspect=0.05)
+ax = fig.add_axes([b, 10*b, 1-2*b, 1-10*b], frameon=False, aspect=0.05)
 
 cmap = plt.get_cmap("Oranges")
-norm = mpl.colors.Normalize(vmin=0,vmax=1)
+norm = mpl.colors.Normalize(vmin=0, vmax=1)
 sm = plt.cm.ScalarMappable(cmap=cmap, norm=norm)
 sm.set_array([])
-plt.colorbar(sm, cax=ax, ticks=np.linspace(0,1,11),
+plt.colorbar(sm, cax=ax, ticks=np.linspace(0, 1, 11),
              orientation="horizontal")
 
 plt.savefig("../figures/colorbar.pdf")

scripts/colormaps.py

@@ -4,7 +4,7 @@
 
 figsize = 4.0, 0.25
 fig = plt.figure(figsize=figsize)
-ax = fig.add_axes([0,0,1,1], frameon=False, aspect=1)
+ax = fig.add_axes([0, 0, 1, 1], frameon=False, aspect=1)
 ymin, ymax=  0, 1
 xmin, xmax = 0, figsize[0]/figsize[1]
 
@@ -13,23 +13,23 @@
 cmaps = ('viridis', 'plasma', 'inferno', 'magma', 'cividis',
 
          'PRGn', 'PiYG', 'RdYlGn', 'BrBG', 'RdGy', 'PuOr', 'RdBu',
-         'RdYlBu',  'Spectral', 'coolwarm', 'bwr', 'seismic',
+         'RdYlBu', 'Spectral', 'coolwarm', 'bwr', 'seismic',
 
          'tab10', 'tab20', 'tab20b', 'tab20c',
          'Pastel1', 'Pastel2', 'Paired',
          'Set1', 'Set2', 'Set3', 'Accent', 'Dark2',
 
-         'Greys', 'Reds','Oranges', 'YlOrBr', 'YlOrRd', 'OrRd',
+         'Greys', 'Reds', 'Oranges', 'YlOrBr', 'YlOrRd', 'OrRd',
          'PuRd', 'RdPu', 'BuPu', 'Purples', 'YlGnBu', 'Blues',
-         'PuBu', 'GnBu', 'PuBuGn', 'BuGn','Greens', 'YlGn',
+         'PuBu', 'GnBu', 'PuBuGn', 'BuGn', 'Greens', 'YlGn',
 
          'bone', 'gray', 'pink', 'afmhot', 'hot', 'gist_heat', 'copper',
          'Wistia', 'autumn', 'summer', 'spring', 'cool', 'winter',
 
          'twilight', 'twilight_shifted', 'hsv',
 
          'terrain', 'ocean', 'gist_earth', 'cubehelix', 'rainbow'
-)
+         )
 
 for cmap in cmaps:
     n = 512
@@ -40,7 +40,7 @@
     if cmap in ['Pastel2', 'Accent', 'Dark2', 'Set2']: n = 8
     if cmap in ['Set3']: n = 12
     if cmap in ['Greys']: n = 11
-    Z = np.linspace(0,1,n).reshape(1,n)
+    Z = np.linspace(0, 1, n).reshape(1, n)
 
     ax.imshow(Z, extent=[xmin, xmax, ymin, ymax], cmap=plt.get_cmap(cmap))
     ax.set_xlim(xmin, xmax), ax.set_xticks([])

scripts/colors.py

@@ -9,7 +9,7 @@
 
 figsize = 4.0, 0.25
 fig = plt.figure(figsize=figsize)
-ax = fig.add_axes([0,0,1,1], frameon=False, aspect=1)
+ax = fig.add_axes([0, 0, 1, 1], frameon=False, aspect=1)
 ymin, ymax=  0, 1
 xmin, xmax = 0, figsize[0]/figsize[1]
 ax.set_xlim(xmin, xmax), ax.set_xticks([])
@@ -19,22 +19,22 @@
 # -----------------------------------------------------------------------------
 palettes = {
     'raw' : ['b', 'g', 'r', 'c', 'm', 'y', 'k', 'w'],
-    'rgba' : [(1,0,0), (1,0,0,0.75), (1,0,0,0.50), (1,0,0,0.25)],
+    'rgba' : [(1, 0, 0), (1, 0, 0, 0.75), (1, 0, 0, 0.50), (1, 0, 0, 0.25)],
     'HexRGBA' : ["#FF0000", "#FF0000BB", "#FF000088", "#FF000044"],
     'cycle' : ["C%d" % i for i in range(10)],
     'grey' : ["%1.1f" % (i/10) for i in range(11)],
     'name' : ["DarkRed", "Firebrick", "Crimson", "IndianRed", "Salmon" ] }
 
 for name, colors in palettes.items():
-    C = mpl.colors.to_rgba_array(colors).reshape((1,len(colors),4))
+    C = mpl.colors.to_rgba_array(colors).reshape((1, len(colors), 4))
     ax.imshow(C, extent=[xmin, xmax, ymin, ymax])
     dx = (xmax-xmin)/len(colors)
     for i in range(len(colors)):
         color = "white"
         if colors[i] in ['1.0', 'w']: color = "black"
         text = str(colors[i]).replace(' ', '')
         ax.text((i+0.5)*dx, (ymin+ymax)/2, text, color=color, zorder=10,
-                family = "Source Code Pro", size=9, ha="center", va="center")
+                family="Source Code Pro", size=9, ha="center", va="center")
 
     plt.savefig("../figures/colors-%s.pdf" % name)
     ax.clear()