Added legend section

Author: rougier

handout-intermediate.tex

@@ -36,7 +36,7 @@
   %
   emph = { plot, scatter, imshow, bar, contourf, pie, subplots, spines,
            add_gridspec, add_subplot, set_xscale, set_minor_locator,
-           set_minor_formatter, tick_params, fill_betweenx, text,
+           set_minor_formatter, tick_params, fill_betweenx, text, legend,
            errorbar, boxplot, hist, title, xlabel, ylabel, suptitle },
   emphstyle = {\ttfamily\bfseries}
 }
@@ -78,7 +78,7 @@ \section*{\LARGE \rmfamily
 
 \includegraphics[width=\linewidth]{anatomy-cropped.pdf}
 
-\subsection*{Figure, axes \& spines}
+\subsection*{\rmfamily Figure, axes \& spines}
 
 % -----------------------------------------------------------------------------
 \begin{tabular}{@{}m{.821\linewidth}m{.169\linewidth}}
@@ -113,7 +113,7 @@ \subsection*{Figure, axes \& spines}
 
 
 % -----------------------------------------------------------------------------
-\subsection*{Ticks \& labels}
+\subsection*{\rmfamily Ticks \& labels}
 
 \begin{lstlisting}[basicstyle=\ttfamily\small]
  from mpl.ticker import MultipleLocator as ML
@@ -125,7 +125,7 @@ \subsection*{Ticks \& labels}
 \includegraphics[width=\linewidth]{tick-multiple-locator.pdf}
 
 % -----------------------------------------------------------------------------
-\subsection*{Lines \& markers}
+\subsection*{\rmfamily Lines \& markers}
 
 \begin{lstlisting}
  X = np.linspace(0.1, 10*np.pi, 1000)
@@ -135,7 +135,7 @@ \subsection*{Lines \& markers}
 \includegraphics[width=\linewidth]{sine-marker.pdf}
 
 % -----------------------------------------------------------------------------
-\subsection*{Scales \& Projections}
+\subsection*{\rmfamily Scales \& Projections}
 
 \begin{lstlisting}
  fig, ax = plt.subplots()
@@ -144,7 +144,7 @@ \subsection*{Scales \& Projections}
 \end{lstlisting}
 \includegraphics[width=\linewidth]{sine-logscale.pdf}
 
-\subsection*{Text \& Ornaments}
+\subsection*{\rmfamily Text \& Ornaments}
 \begin{lstlisting}[]
  ax.fill_betweenx([-1,1],[0],[2*np.pi])
  ax.text(0, -1, r" Period $\Phi$")
@@ -153,12 +153,33 @@ \subsection*{Text \& Ornaments}
 
 
 % -----------------------------------------------------------------------------
-\subsection*{Size \& DPI}
+\subsection*{\rmfamily Legend}
+\begin{lstlisting}[]
+ ax.plot(X, np.sin(X), "C0", label="Sine")
+ ax.plot(X, np.cos(X), "C1", label="Cosine")
+ ax.legend(bbox_to_anchor=(0,1,1,.1),ncol=2,
+           mode="expand",  loc="lower left")
+\end{lstlisting}
+\includegraphics[width=\linewidth]{sine-legend.pdf}
+
+
+% -----------------------------------------------------------------------------
+\subsection*{\rmfamily Colors}
+
+Any color can be used but Matplotlib offers a set of colors:\\
+\includegraphics[width=\linewidth]{colors-cycle.pdf} \smallskip
+\includegraphics[width=\linewidth]{colors-grey.pdf}\\
+As well as nice colormaps (viridis an magma):\\
+\includegraphics[width=\linewidth]{colormap-viridis.pdf} \smallskip
+\includegraphics[width=\linewidth]{colormap-magma.pdf} \medskip
+
+% -----------------------------------------------------------------------------
+\subsection*{\rmfamily Size \& DPI}
 
 Consider a square figure to be included in a two-columns A4 paper with
 2cm margins on each side and a column separation of 1cm. The width of
 a figure is (21 - 2*2 - 1)/2 = 8cm. One inch being 2.54cm, figure size
-should be 3.15$\times$3.15 inches.
+should be 3.15$\times$3.15 in.
 \begin{lstlisting}[]
  fig = plt.figure(figsize=(3.15,3.15), dpi=50)
  plt.savefig("figure.pdf", dpi=600)

scripts/sine.py

@@ -31,26 +31,27 @@
 # plt.savefig("../figures/sine-logscale.pdf", dpi=100)
 
 
+# fig = plt.figure(figsize=(7,1.5))
+# plt.plot(X, Y, "C1", lw=2)
+# plt.fill_betweenx([-1.5,1.5],[0],[2*np.pi], color=".9")
+# plt.text(0, -1, r" Period $\Phi$")
+# # plt.xticks([]), plt.yticks([])
+# plt.ylim(-1.5, 1.5)
+# plt.tight_layout()
+# plt.savefig("../figures/sine-period.pdf", dpi=100)
+# plt.show()
+
+
 fig = plt.figure(figsize=(7,1.5))
-plt.plot(X, Y, "C1", lw=2)
-plt.fill_betweenx([-1.5,1.5],[0],[2*np.pi], color=".9")
-plt.text(0, -1, r" Period $\Phi$")
-# plt.xticks([]), plt.yticks([])
-plt.ylim(-1.5, 1.5)
+plt.plot(X, np.sin(X), "C0", lw=2, label="Sine")
+plt.plot(X, np.cos(X), "C1", lw=2, label="Cosine")
+plt.legend(bbox_to_anchor = (0.0, .9, 1.02, 0.1),
+           frameon=False, mode="expand", ncol=2, loc="lower left")
+plt.title("Sine and Cosine")
+plt.xticks([]), plt.yticks([])
+plt.ylim(-1.25, 1.25)
 plt.tight_layout()
-plt.savefig("../figures/sine-period.pdf", dpi=100)
+plt.savefig("../figures/sine-legend.pdf", dpi=100)
 plt.show()
 
-# X = np.linspace(0, 2*np.pi, 10000)
-# Y = np.sin(10*X)
 
-# fig = plt.figure(figsize=(3,3))
-# ax = plt.subplot(projection="polar")
-# ax.set_rorigin(-3)
-# ax.set_xticks(np.linspace(0, 2*np.pi, 8+1 ))
-# ax.set_xticklabels([])
-# ax.set_yticklabels([])
-# plt.ylim(-1.5, 1.5)
-# plt.plot(X, Y, "C1")
-# plt.savefig("../figures/sine-polar.pdf", dpi=100)
-# plt.show()