13 lines
371 B
Python
13 lines
371 B
Python
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 |