ENH: expose the gallery in the statistics and matplotlib chapters

Author: GaelVaroquaux

intro/matplotlib/examples/README.txt

@@ -1,3 +1,7 @@
-Examples for the Matplotlib chapter
-===================================
+
+
+Code samples for Matplotlib
+----------------------------
+
+
 

intro/matplotlib/examples/plot_bar.py

@@ -1,44 +1,31 @@
 """
-Bar plot advanced
-==================
+Bar plots
+==========
 
-An more elaborate bar plot example
+An example of bar plots with matplotlib.
 """
 
 import numpy as np
 import matplotlib.pyplot as plt
 
-n = 16
+n = 12
 X = np.arange(n)
 Y1 = (1 - X / float(n)) * np.random.uniform(0.5, 1.0, n)
 Y2 = (1 - X / float(n)) * np.random.uniform(0.5, 1.0, n)
-plt.bar(X, Y1, facecolor='#9999ff', edgecolor='white')
+
+plt.axes([0.025, 0.025, 0.95, 0.95])
+plt.bar(X, +Y1, facecolor='#9999ff', edgecolor='white')
 plt.bar(X, -Y2, facecolor='#ff9999', edgecolor='white')
+
+for x, y in zip(X, Y1):
+    plt.text(x + 0.4, y + 0.05, '%.2f' % y, ha='center', va= 'bottom')
+
+for x, y in zip(X, Y2):
+    plt.text(x + 0.4, -y - 0.05, '%.2f' % y, ha='center', va= 'top')
+
 plt.xlim(-.5, n)
 plt.xticks(())
-plt.ylim(-1, 1)
+plt.ylim(-1.25, 1.25)
 plt.yticks(())
 
-
-# Add a title and a box around it
-from matplotlib.patches import FancyBboxPatch
-ax = plt.gca()
-ax.add_patch(FancyBboxPatch((-0.05, .87),
-                            width=.66, height=.165, clip_on=False,
-                            boxstyle="square,pad=0", zorder=3,
-                            facecolor='white', alpha=1.0,
-                            transform=plt.gca().transAxes))
-
-plt.text(-0.05, 1.02, " Bar Plot:              plt.bar(...)\n",
-      horizontalalignment='left',
-      verticalalignment='top',
-      size='xx-large',
-      transform=plt.gca().transAxes)
-
-plt.text(-0.05, 1.01, "\n\n   Make a bar plot with rectangles ",
-      horizontalalignment='left',
-      verticalalignment='top',
-      size='large',
-      transform=plt.gca().transAxes)
-
 plt.show()

intro/matplotlib/examples/plot_bar_ex.py

@@ -1,49 +1,28 @@
 """
-Display the contours of a function
-===================================
-
-An example demoing how to plot the contours of a function, with
-additional layout tweeks.
+Displaying the contours of a function
+======================================
 
+An example showing how to display the contours of a function with
+matplotlib.
 """
 
 import numpy as np
 import matplotlib.pyplot as plt
 
 def f(x,y):
-    return (1 - x / 2 + x ** 5 + y ** 3) * np.exp(-x ** 2 - y ** 2)
+    return (1 - x / 2 + x**5 + y**3) * np.exp(-x**2 -y**2)
 
 n = 256
 x = np.linspace(-3, 3, n)
 y = np.linspace(-3, 3, n)
-X, Y = np.meshgrid(x, y)
+X,Y = np.meshgrid(x, y)
+
+plt.axes([0.025, 0.025, 0.95, 0.95])
 
 plt.contourf(X, Y, f(X, Y), 8, alpha=.75, cmap=plt.cm.hot)
-C = plt.contour(X, Y, f(X,Y), 8, colors='black', linewidth=.5)
+C = plt.contour(X, Y, f(X, Y), 8, colors='black', linewidth=.5)
 plt.clabel(C, inline=1, fontsize=10)
+
 plt.xticks(())
 plt.yticks(())
-
-
-# Add a title and a box around it
-from matplotlib.patches import FancyBboxPatch
-ax = plt.gca()
-ax.add_patch(FancyBboxPatch((-0.05, .87),
-                            width=.66, height=.165, clip_on=False,
-                            boxstyle="square,pad=0", zorder=3,
-                            facecolor='white', alpha=1.0,
-                            transform=plt.gca().transAxes))
-
-plt.text(-0.05, 1.02, " Contour Plot: plt.contour(..)\n",
-      horizontalalignment='left',
-      verticalalignment='top',
-      size='xx-large',
-      transform=plt.gca().transAxes)
-
-plt.text(-0.05, 1.01, "\n\n  Draw contour lines and filled contours ",
-      horizontalalignment='left',
-      verticalalignment='top',
-      size='large',
-      transform=plt.gca().transAxes)
-
 plt.show()

intro/matplotlib/examples/plot_boxplot.py

@@ -1,48 +1,25 @@
 """
-Grid elaborate
-===============
+Grid
+====
 
-An example displaying a grid on the axes and tweaking the layout.
+Displaying a grid on the axes in matploblib.
 """
 
 import matplotlib.pyplot as plt
-from matplotlib.ticker import MultipleLocator
-
-fig = plt.figure(figsize=(8, 6), dpi=72, facecolor="white")
-axes = plt.subplot(111)
-axes.set_xlim(0, 4)
-axes.set_ylim(0, 3)
-
-axes.xaxis.set_major_locator(MultipleLocator(1.0))
-axes.xaxis.set_minor_locator(MultipleLocator(0.1))
-axes.yaxis.set_major_locator(MultipleLocator(1.0))
-axes.yaxis.set_minor_locator(MultipleLocator(0.1))
-axes.grid(which='major', axis='x', linewidth=0.75, linestyle='-', color='0.75')
-axes.grid(which='minor', axis='x', linewidth=0.25, linestyle='-', color='0.75')
-axes.grid(which='major', axis='y', linewidth=0.75, linestyle='-', color='0.75')
-axes.grid(which='minor', axis='y', linewidth=0.25, linestyle='-', color='0.75')
-axes.set_xticklabels([])
-axes.set_yticklabels([])
-
-
-# Add a title and a box around it
-from matplotlib.patches import FancyBboxPatch
-ax = plt.gca()
-ax.add_patch(FancyBboxPatch((-0.05, .87),
-                            width=.66, height=.165, clip_on=False,
-                            boxstyle="square,pad=0", zorder=3,
-                            facecolor='white', alpha=1.0,
-                            transform=plt.gca().transAxes))
-
-plt.text(-0.05, 1.02, " Grid:                  plt.grid(...)\n",
-          horizontalalignment='left',
-          verticalalignment='top',
-          size='xx-large',
-          transform=axes.transAxes)
-
-plt.text(-0.05, 1.01, "\n\n    Draw ticks and grid ",
-          horizontalalignment='left',
-          verticalalignment='top',
-          size='large',
-          transform=axes.transAxes)
 
+ax = plt.axes([0.025, 0.025, 0.95, 0.95])
+
+ax.set_xlim(0,4)
+ax.set_ylim(0,3)
+ax.xaxis.set_major_locator(plt.MultipleLocator(1.0))
+ax.xaxis.set_minor_locator(plt.MultipleLocator(0.1))
+ax.yaxis.set_major_locator(plt.MultipleLocator(1.0))
+ax.yaxis.set_minor_locator(plt.MultipleLocator(0.1))
+ax.grid(which='major', axis='x', linewidth=0.75, linestyle='-', color='0.75')
+ax.grid(which='minor', axis='x', linewidth=0.25, linestyle='-', color='0.75')
+ax.grid(which='major', axis='y', linewidth=0.75, linestyle='-', color='0.75')
+ax.grid(which='minor', axis='y', linewidth=0.25, linestyle='-', color='0.75')
+ax.set_xticklabels([])
+ax.set_yticklabels([])
+
+plt.show()

intro/matplotlib/examples/plot_contour.py

@@ -1,46 +1,26 @@
 """
-Imshow demo
-============
+Imshow elaborate
+=================
 
-Demoing imshow
+An example demoing imshow and styling the figure.
 """
 
 import numpy as np
 import matplotlib.pyplot as plt
 
 def f(x, y):
-    return (1 - x / 2 + x ** 5 + y ** 3) * np.exp(-x ** 2 - y ** 2)
+    return (1 - x / 2 + x ** 5 + y ** 3 ) * np.exp(-x ** 2 - y ** 2)
 
 n = 10
-x = np.linspace(-3, 3, 8 * n)
-y = np.linspace(-3, 3, 6 * n)
+x = np.linspace(-3, 3, 3.5 * n)
+y = np.linspace(-3, 3, 3.0 * n)
 X, Y = np.meshgrid(x, y)
 Z = f(X, Y)
+
+plt.axes([0.025, 0.025, 0.95, 0.95])
 plt.imshow(Z, interpolation='nearest', cmap='bone', origin='lower')
+plt.colorbar(shrink=.92)
+
 plt.xticks(())
 plt.yticks(())
-
-
-# Add a title and a box around it
-from matplotlib.patches import FancyBboxPatch
-ax = plt.gca()
-ax.add_patch(FancyBboxPatch((-0.05, .87),
-                            width=.66, height=.165, clip_on=False,
-                            boxstyle="square,pad=0", zorder=3,
-                            facecolor='white', alpha=1.0,
-                            transform=plt.gca().transAxes))
-
-plt.text(-0.05, 1.02, " Imshow:       plt.imshow(...)\n",
-        horizontalalignment='left',
-        verticalalignment='top',
-        size='xx-large',
-        transform=plt.gca().transAxes)
-
-plt.text(-0.05, 1.01, "\n\n   Display an image to current axes ",
-        horizontalalignment='left',
-        verticalalignment='top',
-        family='Lint McCree Intl BB',
-        size='large',
-        transform=plt.gca().transAxes)
-
 plt.show()