| Download
All published worksheets from http://sagenb.org
Project: sagenb.org published worksheets
Views: 168742Image: ubuntu2004
[-0.23176471 -3.97176471 4.18 -0.01176471]
array([ 2., 3., 4., 5.])
array([[ 1, 2, 3],
[35, 23, 6],
[15, 13, 16]])
array([2, 0, 2])
File: /sagenb/sage_install/sage-4.7/local/lib/python2.6/site-packages/numpy/core/fromnumeric.py
Type: <type ‘function’>
Definition: np.argmax(a, axis=None)
Docstring:
Indices of the maximum values along an axis.
- a : array_like
- Input array.
- axis : int, optional
- By default, the index is into the flattened array, otherwise along the specified axis.
- index_array : ndarray of ints
- Array of indices into the array. It has the same shape as a.shape with the dimension along axis removed.
ndarray.argmax, argmin amax : The maximum value along a given axis. unravel_index : Convert a flat index into an index tuple.
In case of multiple occurrences of the maximum values, the indices corresponding to the first occurrence are returned.
>>> a = np.arange(6).reshape(2,3) >>> a array([[0, 1, 2], [3, 4, 5]]) >>> np.argmax(a) 5 >>> np.argmax(a, axis=0) array([1, 1, 1]) >>> np.argmax(a, axis=1) array([2, 2])>>> b = np.arange(6) >>> b[1] = 5 >>> b array([0, 5, 2, 3, 4, 5]) >>> np.argmax(b) # Only the first occurrence is returned. 1