Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #!/usr/bin/env python
- #-*- coding:utf-8 -*-
- import sys
- import timeit
- def main():
- test(2000000, "range")
- test(2000000, "xrange")
- test(2000000, "numpy.arange", "import numpy")
- test2(2000000)
- test3(2000000)
- def test(n, fxn, setup='pass'):
- s = """\
- nrepArray = %s(%d)
- for i in %s(%d):
- pass
- """ % (fxn, n, fxn, n)
- t = timeit.Timer(stmt=s, setup=setup)
- time = t.timeit(number=1)
- print "%s took %f seconds (n=%d)" % (fxn, time, n)
- def test2(n):
- s = """[x * 2 for x in xrange(%d)]""" % n
- t = timeit.Timer(stmt=s)
- time = t.timeit(number=1)
- print "scale multiplication (xrange) took %f seconds (n=%d)" % (time, n)
- def test3(n):
- s = """numpy.arange(%d) * 2""" % n
- t = timeit.Timer(stmt=s, setup="import numpy")
- time = t.timeit(number=1)
- print "scale multiplication (numpy.arange) took %f seconds (n=%d)" % (time, n)
- if __name__ == '__main__':
- sys.exit(main())
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement