Spaces:
Running
Running
| #!/usr/bin/env python3 | |
| # -*- coding: utf-8 -*- | |
| import timeit | |
| import numpy | |
| ############################################################################### | |
| # Global variables # | |
| ############################################################################### | |
| # Small arrays | |
| xs = numpy.random.uniform(-1, 1, 6).reshape(2, 3) | |
| ys = numpy.random.uniform(-1, 1, 6).reshape(2, 3) | |
| zs = xs + 1j * ys | |
| m1 = [[True, False, False], [False, False, True]] | |
| m2 = [[True, False, True], [False, False, True]] | |
| nmxs = numpy.ma.array(xs, mask=m1) | |
| nmys = numpy.ma.array(ys, mask=m2) | |
| nmzs = numpy.ma.array(zs, mask=m1) | |
| # Big arrays | |
| xl = numpy.random.uniform(-1, 1, 100*100).reshape(100, 100) | |
| yl = numpy.random.uniform(-1, 1, 100*100).reshape(100, 100) | |
| zl = xl + 1j * yl | |
| maskx = xl > 0.8 | |
| masky = yl < -0.8 | |
| nmxl = numpy.ma.array(xl, mask=maskx) | |
| nmyl = numpy.ma.array(yl, mask=masky) | |
| nmzl = numpy.ma.array(zl, mask=maskx) | |
| ############################################################################### | |
| # Functions # | |
| ############################################################################### | |
| def timer(s, v='', nloop=500, nrep=3): | |
| units = ["s", "ms", "µs", "ns"] | |
| scaling = [1, 1e3, 1e6, 1e9] | |
| print("%s : %-50s : " % (v, s), end=' ') | |
| varnames = ["%ss,nm%ss,%sl,nm%sl" % tuple(x*4) for x in 'xyz'] | |
| setup = 'from __main__ import numpy, ma, %s' % ','.join(varnames) | |
| Timer = timeit.Timer(stmt=s, setup=setup) | |
| best = min(Timer.repeat(nrep, nloop)) / nloop | |
| if best > 0.0: | |
| order = min(-int(numpy.floor(numpy.log10(best)) // 3), 3) | |
| else: | |
| order = 3 | |
| print("%d loops, best of %d: %.*g %s per loop" % (nloop, nrep, | |
| 3, | |
| best * scaling[order], | |
| units[order])) | |
| def compare_functions_1v(func, nloop=500, | |
| xs=xs, nmxs=nmxs, xl=xl, nmxl=nmxl): | |
| funcname = func.__name__ | |
| print("-"*50) | |
| print(f'{funcname} on small arrays') | |
| module, data = "numpy.ma", "nmxs" | |
| timer("%(module)s.%(funcname)s(%(data)s)" % locals(), v="%11s" % module, nloop=nloop) | |
| print("%s on large arrays" % funcname) | |
| module, data = "numpy.ma", "nmxl" | |
| timer("%(module)s.%(funcname)s(%(data)s)" % locals(), v="%11s" % module, nloop=nloop) | |
| return | |
| def compare_methods(methodname, args, vars='x', nloop=500, test=True, | |
| xs=xs, nmxs=nmxs, xl=xl, nmxl=nmxl): | |
| print("-"*50) | |
| print(f'{methodname} on small arrays') | |
| data, ver = f'nm{vars}l', 'numpy.ma' | |
| timer("%(data)s.%(methodname)s(%(args)s)" % locals(), v=ver, nloop=nloop) | |
| print("%s on large arrays" % methodname) | |
| data, ver = "nm%sl" % vars, 'numpy.ma' | |
| timer("%(data)s.%(methodname)s(%(args)s)" % locals(), v=ver, nloop=nloop) | |
| return | |
| def compare_functions_2v(func, nloop=500, test=True, | |
| xs=xs, nmxs=nmxs, | |
| ys=ys, nmys=nmys, | |
| xl=xl, nmxl=nmxl, | |
| yl=yl, nmyl=nmyl): | |
| funcname = func.__name__ | |
| print("-"*50) | |
| print(f'{funcname} on small arrays') | |
| module, data = "numpy.ma", "nmxs,nmys" | |
| timer("%(module)s.%(funcname)s(%(data)s)" % locals(), v="%11s" % module, nloop=nloop) | |
| print(f'{funcname} on large arrays') | |
| module, data = "numpy.ma", "nmxl,nmyl" | |
| timer("%(module)s.%(funcname)s(%(data)s)" % locals(), v="%11s" % module, nloop=nloop) | |
| return | |
| if __name__ == '__main__': | |
| compare_functions_1v(numpy.sin) | |
| compare_functions_1v(numpy.log) | |
| compare_functions_1v(numpy.sqrt) | |
| compare_functions_2v(numpy.multiply) | |
| compare_functions_2v(numpy.divide) | |
| compare_functions_2v(numpy.power) | |
| compare_methods('ravel', '', nloop=1000) | |
| compare_methods('conjugate', '', 'z', nloop=1000) | |
| compare_methods('transpose', '', nloop=1000) | |
| compare_methods('compressed', '', nloop=1000) | |
| compare_methods('__getitem__', '0', nloop=1000) | |
| compare_methods('__getitem__', '(0,0)', nloop=1000) | |
| compare_methods('__getitem__', '[0,-1]', nloop=1000) | |
| compare_methods('__setitem__', '0, 17', nloop=1000, test=False) | |
| compare_methods('__setitem__', '(0,0), 17', nloop=1000, test=False) | |
| print("-"*50) | |
| print("__setitem__ on small arrays") | |
| timer('nmxs.__setitem__((-1,0),numpy.ma.masked)', 'numpy.ma ', nloop=10000) | |
| print("-"*50) | |
| print("__setitem__ on large arrays") | |
| timer('nmxl.__setitem__((-1,0),numpy.ma.masked)', 'numpy.ma ', nloop=10000) | |
| print("-"*50) | |
| print("where on small arrays") | |
| timer('numpy.ma.where(nmxs>2,nmxs,nmys)', 'numpy.ma ', nloop=1000) | |
| print("-"*50) | |
| print("where on large arrays") | |
| timer('numpy.ma.where(nmxl>2,nmxl,nmyl)', 'numpy.ma ', nloop=100) | |