B <ÓFdã @sÊddlmZmZddlZddlZej ddd¡ dd¡Zej ddd¡ dd¡Z ede Z d d d gd d d ggZd d d gd d d ggZej jeedZej je edZej je edZej ddd¡ d d ¡Zej ddd¡ d d ¡ZedeZedkZedkZej jeedZej jeedZej jeedZd5dd„Zdeeeefdd„Zddd eeeefdd„Zdd eee eeeeef dd„ZedkrÆeejƒeej ƒeej!ƒeej"ƒeej#ƒeej$ƒeddddeddd dded!ddded"ddded#d$dded#d%dded#d&dded'd(dd d)ed'd*dd d)e%d+ƒe%d,ƒed-d.dde%d+ƒe%d/ƒed0d.dde%d+ƒe%d1ƒed2d.dde%d+ƒe%d3ƒed4d.d ddS)6é)ÚdivisionÚprint_functionNéÿÿÿÿééééyð?TF)Úmaski'édgš™™™™™é?gš™™™™™é¿ÚéôcCs¾ddddg}ddddg}td ||fd ddd „dDƒ}dd |¡}tj||d}t| ||¡ƒ|} | dkr”ttt t | ¡¡dƒdƒ} nd} td||d| || || fƒdS)NÚsZmsuÂµsÚnsrg@@g€„.AgeÍÍAz %s : %-50s : ú )ÚendcSsg|]}dt|dƒ‘qS)z%ss,nm%ss,%sl,nm%slé)Útuple)Ú.0Úx©rúA/opt/alt/python37/lib64/python3.7/site-packages/numpy/ma/bench.pyú *sztimer..Zxyzz"from __main__ import numpy, ma, %sú,)ZstmtÚsetupgrz&%d loops, best of %d: %.*g %s per loop) ÚprintÚjoinÚtimeitÚTimerÚminÚrepeatÚintÚnumpyZfloorZlog10)r ÚvÚnloopZnrepZunitsZscalingZvarnamesrrZbestÚorderrrrÚtimer&s" r%c Csj|j}tdƒtd|ƒd\}}tdtƒd||dtd|ƒd\}}tdtƒd||ddS) Nz2--------------------------------------------------z%s on small arrays)znumpy.maÚnmxsz!%(module)s.%(funcname)s(%(data)s)z%11s)r"r#z%s on large arrays)znumpy.maÚnmxl)Ú__name__rr%Úlocals) Úfuncr#Úxsr&Úxlr'ÚfuncnameÚmoduleÚdatarrrÚcompare_functions_1v8sr0rc Cshtdƒtd|ƒd|d} } tdtƒ| |dtd|ƒd|d} } tdtƒ| |ddS) Nz2--------------------------------------------------z%s on small arraysznm%ssznumpy.maz!%(data)s.%(methodname)s(%(args)s))r"r#z%s on large arraysznm%sl)rr%r))Z methodnameÚargsÚvarsr#Útestr+r&r,r'r/ZverrrrÚcompare_methodsEsr4cCsj|j}tdƒtd|ƒd\}} tdtƒd||dtd|ƒd\}} tdtƒd||ddS) Nz2--------------------------------------------------z%s on small arrays)znumpy.maz nmxs,nmysz!%(module)s.%(funcname)s(%(data)s)z%11s)r"r#z%s on large arrays)znumpy.maz nmxl,nmyl)r(rr%r))r*r#r3r+r&ÚysÚnmysr,r'ÚylÚnmylr-r.r/rrrÚcompare_functions_2vQsr9Ú__main__Zraveliè)r#Ú conjugateÚzZ transposeZ compressedÚ__getitem__Ú0z(0,0)z[0,-1]Ú__setitem__z0, 17)r#r3z (0,0), 17z2--------------------------------------------------z__setitem__ on small arraysz(nmxs.__setitem__((-1,0),numpy.ma.masked)znumpy.ma z__setitem__ on large arraysz(nmxl.__setitem__((-1,0),numpy.ma.masked)zwhere on small arraysz numpy.ma.where(nmxs>2,nmxs,nmys)zwhere on large arraysz numpy.ma.where(nmxl>2,nmxl,nmyl))rrr)&Z __future__rrrr!ZrandomZuniformZreshaper+r5ZzsZm1Zm2ZmaZarrayr&r6Znmzsr,r7ZzlZmaskxZmaskyr'r8Znmzlr%r0r4r9r(ZsinÚlogZsqrtZmultiplyZdivideZpowerrrrrrÚsp