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
"""Numeric module defining a multi-dimensional array and useful procedures for
