[Python]Numpyの配列(Array)の行ごとの最大値を取得する(get rows max value)には？

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

Numpyの配列(Array)の行ごとの最大値を取得する方法を紹介します。

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

方法

Numpyの配列(Array)の行ごとの最大値を取得する方法は、２つあります。

max()メソッド

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

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

max()メソッドの引数「axis」に「1」を指定します。

array.max(axis=1)

引数「axis」に「１」を指定したmax()メソッドは、呼び出したNumpy配列の行ごとの最大値を返します。

使用例

import numpy as np

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

max_nums = nums.max(axis=1) 

print(max_nums) #[ 100   10 1000]

numpy.amax()

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

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

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

np.amax(array, axis=1)

上記のamax()関数は、引数に指定したNumpy配列の行ごとの最大値を返します。

使用例

import numpy as np

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

max_nums = np.amax(nums, axis=1)

print(max_nums) #[ 100   10 1000]