numba routines
This commit is contained in:
parent
1a8afb97d0
commit
5a88eebf4a
|
|
@ -0,0 +1,13 @@
|
|||
import numba
|
||||
import numpy as np
|
||||
|
||||
@numba.jit(nopython=True)
|
||||
def minmax(array):
|
||||
N = array.shape[0]
|
||||
vmin = np.full((N,), np.inf,dtype=array.dtype)
|
||||
vmax = np.full((N,),-np.inf,dtype=array.dtype)
|
||||
for ii in range(N):
|
||||
for val in array[ii,...].ravel():
|
||||
vmin[ii] = min(vmin[ii],val)
|
||||
vmax[ii] = max(vmax[ii],val)
|
||||
return vmin,vmax
|
||||
Loading…
Reference in New Issue