python - 1 つの軸に沿って NumPy 配列の最大要素のインデックスを取得する方法

Question

私は2次元のNumPy配列を持っています。軸の最大値を取得する方法を知っています:

>>> a = array([[1,2,3],[4,3,1]])
>>> amax(a,axis=0)
array([4, 3, 3])

最大要素のインデックスを取得するにはどうすればよいですか? array([1,1,0])代わりに出力したいと思います。

Answer 1

>>> a.argmax(axis=0)

array([1, 1, 0])

Answer 2

>>> import numpy as np
>>> a = np.array([[1,2,3],[4,3,1]])
>>> i,j = np.unravel_index(a.argmax(), a.shape)
>>> a[i,j]
4

Answer 3

v = alli.max()
index = alli.argmax()
x, y = index/8, index%8

Answer 4

numpy 配列の最小値と最大値のインデックスをそれぞれ返すことがあり、それによってargmin()提供argmax()されます。numpy

たとえば、1次元配列の場合、次のようにします

import numpy as np

a = np.array([50,1,0,2])

print(a.argmax()) # returns 0
print(a.argmin()) # returns 2

同様に、多次元配列についても

import numpy as np

a = np.array([[0,2,3],[4,30,1]])

print(a.argmax()) # returns 4
print(a.argmin()) # returns 0

これらは、最初に出現したインデックスのみを返すことに注意してください。