[Python]Numpyの配列(Array)の最大値のインデックスを取得するには？

どうも、ちょげ(@chogetarou)です。

Numpyの配列(Array)の最大値のインデックスを取得する方法を紹介します。

方法

Numpyの配列(Array)の最大値のインデックスを取得する方法は、２つあります。

argmax()メソッド

１つは、argmax()メソッドを使う方法です。

具体的には、Numpyの配列からargmax()メソッドを呼び出します。

array.argmax()

argmax()メソッドは、呼び出したNumpy配列の最大値のインデックスを返します。

np.unravel_index(array.argmax(), array.shape)

使用例

import numpy as np

nums = np.array([[1, 2, 3],
                 [4, 10, 6], 
                 [7, 8, 9]])

max_index = nums.argmax()

max_dim_index = np.unravel_index(nums.argmax(), nums.shape)

print(max_index) #4

print(max_dim_index) #(1, 1)

numpy.argmax()

もう１つは、numpyのargmax()を使う方法です。

まず、numpyからargmax()を呼び出します。

そして、argmax()関数の引数にNumpy配列を指定します。

np.argmax(array)

argmax()関数は、引数に指定したNumpy配列の最大値のインデックスを返します。

np.unravel_index(np.argmax(array), array.shape)

使用例

import numpy as np

nums = np.array([[1, 2, 3],
                 [4, 10, 6], 
                 [7, 8, 9]])

max_index = np.argmax(nums)
max_dim_index = np.unravel_index(np.argmax(nums), nums.shape)

print(max_index) #10
print(max_dim_index) #(1, 1)

まとめ

Numpyの配列(Array)の最大値のインデックスを取得する方法は、次の２つです。

• argmax()メソッドを使う方法
• numpy.argmax()を使う方法