Overhaul of Part 1

exercises/1.1-subplots_and_basic_plotting.py

@@ -0,0 +1,11 @@
+import numpy as np
+import matplotlib.pyplot as plt
+
+# Try to reproduce the figure shown in images/exercise_1-1.png
+
+# Our data...
+x = np.linspace(0, 10, 100)
+y1, y2, y3 = np.cos(x), np.cos(x + 1), np.cos(x + 2)
+names = ['Signal 1', 'Signal 2', 'Signal 3']
+
+# Can you figure out what to do next to plot x vs y1, y2, and y3 on one figure?

exercises/1.2-legends_and_scaling.py

@@ -0,0 +1,15 @@
+import numpy as np
+import matplotlib.pyplot as plt
+
+# Try to reproduce the figure shown in images/exercise_1-2.png
+# Here's the data and colors used.
+
+t = np.linspace(0, 2 * np.pi, 150)
+x1, y1 = np.cos(t), np.sin(t)
+x2, y2 = 2 * x1, 2 * y1
+
+colors = ['darkred', 'darkgreen']
+
+# Try to plot the two circles, scale the axes as shown and add a legend
+# Hint: it's easiest to combine `ax.axis(...)` and `ax.margins(...)` to scale
+# the axes

exercises/1.3-spines_ticks_and_subplot_spacing.py

@@ -0,0 +1,9 @@
+import matplotlib.pyplot as plt
+import numpy as np
+
+# Try to reproduce the figure shown in images/exercise_1.3.png
+# This one is a bit trickier!
+
+# Here's the data...
+data = [('dogs', 4, 4), ('frogs', -3, 1), ('cats', 1, 5), ('goldfish', -2, 2)]
+animals, friendliness, popularity = zip(*data)

solutions/1.1-subplots_and_basic_plotting.py

@@ -0,0 +1,14 @@
+import numpy as np
+import matplotlib.pyplot as plt
+
+x = np.linspace(0, 10, 100)
+y1, y2, y3 = np.cos(x), np.cos(x + 1), np.cos(x + 2)
+names = ['Signal 1', 'Signal 2', 'Signal 3']
+
+fig, axes = plt.subplots(nrows=3)
+
+for ax, y, name in zip(axes, [y1, y2, y3], names):
+    ax.plot(x, y, color='black')
+    ax.set(xticks=[], yticks=[], title=name)
+
+plt.show()

solutions/1.2-legends_and_scaling.py

@@ -0,0 +1,18 @@
+import numpy as np
+import matplotlib.pyplot as plt
+
+t = np.linspace(0, 2 * np.pi, 150)
+x1, y1 = np.cos(t), np.sin(t)
+x2, y2 = 2 * x1, 2 * y1
+
+colors = ['darkred', 'darkgreen']
+
+fig, ax = plt.subplots()
+ax.plot(x1, y1, color=colors[0], label='Inner', linewidth=3)
+ax.plot(x2, y2, color=colors[1], label='Outer', linewidth=3)
+ax.legend()
+
+ax.axis('equal')
+ax.margins(0.05)
+
+plt.show()

solutions/1.3-spines_ticks_and_subplot_spacing.py

@@ -0,0 +1,25 @@
+import matplotlib.pyplot as plt
+import numpy as np
+
+data = [('dogs', 4, 4), ('frogs', -3, 1), ('cats', 1, 5), ('goldfish', -2, 2)]
+animals, friendliness, popularity = zip(*data)
+
+
+def plot_and_setup_spines(ax, animals, y, ylabel):
+    x = np.arange(len(animals))
+    ax.bar(x, y, align='center', color='gray')
+    ax.set(xticks=x, xticklabels=animals, ylabel=ylabel)
+
+    ax.xaxis.set_ticks_position('bottom')
+    ax.yaxis.set_ticks_position('left')
+    ax.spines['bottom'].set_position(('data', 0))
+    ax.tick_params(axis='x', direction='inout', length=8)
+    ax.margins(0.05)
+
+fig, axes = plt.subplots(nrows=2)
+fig.subplots_adjust(hspace=0.0)
+
+plot_and_setup_spines(axes[0], animals, friendliness, 'Friendliness')
+plot_and_setup_spines(axes[1], animals, popularity, 'Popularity')
+
+plt.show()