Merge pull request #4734 from ericmjl/mep12-scatter_profile.py

jenshnielsen · jenshnielsen · commit d2f91a11c844 · 2015-07-19T23:29:42.000+01:00
mep12 on scatter_profile.py
diff --git a/examples/pylab_examples/scatter_profile.py b/examples/pylab_examples/scatter_profile.py
@@ -1,6 +1,3 @@
-#!/usr/bin/env python
-# -*- noplot -*-
-
 """
 N       Classic     Base renderer    Ext renderer
 20       0.22           0.14            0.14
@@ -9,16 +6,16 @@
 10000    3.30           1.31            0.53
 50000    19.30          6.53            1.98
 """
-from __future__ import print_function
-import pylab
+from __future__ import print_function  # only needed for python 2.x
+import matplotlib.pyplot as plt
+import numpy as np
 
 import time
 
-
 for N in (20, 100, 1000, 10000, 50000):
     tstart = time.time()
-    x = 0.9*pylab.rand(N)
-    y = 0.9*pylab.rand(N)
-    s = 20*pylab.rand(N)
-    pylab.scatter(x, y, s)
+    x = 0.9*np.random.rand(N)
+    y = 0.9*np.random.rand(N)
+    s = 20*np.random.rand(N)
+    plt.scatter(x, y, s)
     print('%d symbols in %1.2f s' % (N, time.time() - tstart))
