[Python]Numpy配列の行ごとの最小値のインデックスを取得するには？

python

2022.05.15

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

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

方法
1. argmin()メソッド
2. numpy.argmin()
まとめ

方法

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

argmin()メソッド

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

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

argmin()の引数「axis」に「1」を指定します。

array.argmin(axis=1)

引数「axis」に「1」を指定したargmin()メソッドは、呼び出したNumpy配列の行ごとの最小値のインデックスを返します。

使用例

import numpy as np

nums = np.array([[1, 2, -100],
                 [4, -1000, 6], 
                 [7, 8, -10]])

min_num = nums.argmin(axis=1)

print(min_num) #[ 2, 1, 2]

numpy.argmin()

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

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

そして、argmin()関数の第１引数にNumpy配列、引数「axis」に「1」を指定します。

np.argmin(array, axis=1)

引数「axis」に「1」を指定したargmax()関数は、引数に指定したNumpy配列の行ごとの最小値のインデックスを返します。

使用例

import numpy as np

nums = np.array([[1, 2, -100],
                 [4, -1000, 6], 
                 [7, 8, -10]])

min_num = np.argmin(nums, axis=1)

print(min_num) #[ 2, 1, 2]