Merge pull request #4933 from ericmjl/mep12-line_colection2.py

jenshnielsen · jenshnielsen · commit 375b420b1086 · 2015-08-16T09:08:57.000+01:00
MEP12 on line_collection2.py
diff --git a/examples/pylab_examples/line_collection2.py b/examples/pylab_examples/line_collection2.py
@@ -1,34 +1,37 @@
-from pylab import *
+import matplotlib.pyplot as plt
+import numpy as np
 from matplotlib.collections import LineCollection
 
 # In order to efficiently plot many lines in a single set of axes,
 # Matplotlib has the ability to add the lines all at once. Here is a
 # simple example showing how it is done.
 
 N = 50
-x = arange(N)
+x = np.arange(N)
 # Here are many sets of y to plot vs x
 ys = [x + i for i in x]
 
 # We need to set the plot limits, they will not autoscale
-ax = axes()
-ax.set_xlim((amin(x), amax(x)))
-ax.set_ylim((amin(amin(ys)), amax(amax(ys))))
+ax = plt.axes()
+ax.set_xlim((np.amin(x), np.amax(x)))
+ax.set_ylim((np.amin(np.amin(ys)), np.amax(np.amax(ys))))
 
 # colors is sequence of rgba tuples
 # linestyle is a string or dash tuple. Legal string values are
 #          solid|dashed|dashdot|dotted.  The dash tuple is (offset, onoffseq)
 #          where onoffseq is an even length tuple of on and off ink in points.
 #          If linestyle is omitted, 'solid' is used
 # See matplotlib.collections.LineCollection for more information
-line_segments = LineCollection([list(zip(x, y)) for y in ys],  # Make a sequence of x,y pairs
+
+# Make a sequence of x,y pairs
+line_segments = LineCollection([list(zip(x, y)) for y in ys],
                                linewidths=(0.5, 1, 1.5, 2),
                                linestyles='solid')
 line_segments.set_array(x)
 ax.add_collection(line_segments)
-fig = gcf()
+fig = plt.gcf()
 axcb = fig.colorbar(line_segments)
 axcb.set_label('Line Number')
 ax.set_title('Line Collection with mapped colors')
-sci(line_segments)  # This allows interactive changing of the colormap.
-show()
+plt.sci(line_segments)  # This allows interactive changing of the colormap.
+plt.show()
