def argmax(a, axis=-1): """argmax(a,axis=-1) returns the indices to the maximum value of the 1-D arrays along the given axis. """ a = array(a, copy=0) n = len(a.shape) if axis < 0: axis += n if axis < 0 or axis >= n: raise ValueError, "argmax axis argument out of bounds" if axis != n - 1: a = swapaxes(a, axis, n - 1) s = multiarray.argmax(a) if axis != n - 1: s = swapaxes(s, axis, -1) return s
def argmax(a, axis=-1): """argmax(a,axis=-1) returns the indices to the maximum value of the 1-D arrays along the given axis. """ a = array(a, copy=0) n = len(a.shape) if axis < 0: axis += n if axis < 0 or axis >= n: raise ValueError, "argmax axis argument out of bounds" if axis != n-1: a = swapaxes(a, axis, n-1) s = multiarray.argmax(a) if axis != n-1: s = swapaxes(s, axis, -1) return s
def argmin(a, axis=-1): """argmin(a,axis=-1) returns the indices to the minimum value of the 1-D arrays along the given axis. """ arra = array(a, copy=0) type = arra.typecode() num = array(0, type) if type in ['bwu']: num = -array(1, type) a = num - arra n = len(a.shape) if axis < 0: axis += n if axis < 0 or axis >= n: raise ValueError, "argmin axis argument out of bounds" if axis != n - 1: a = swapaxes(a, axis, n - 1) s = multiarray.argmax(a) if axis != n - 1: s = swapaxes(s, axis, -1) return s
def argmin(a, axis=-1): """argmin(a,axis=-1) returns the indices to the minimum value of the 1-D arrays along the given axis. """ arra = array(a,copy=0) type = arra.typecode() num = array(0,type) if type in ['bwu']: num = -array(1,type) a = num-arra n = len(a.shape) if axis < 0: axis += n if axis < 0 or axis >= n: raise ValueError, "argmin axis argument out of bounds" if axis != n-1: a = swapaxes(a, axis, n-1) s = multiarray.argmax(a) if axis != n-1: s = swapaxes(s, axis, -1) return s
"""Numeric module defining a multi-dimensional array and useful procedures for