配列のnp.argmax表示

Question

Pythonを学習している初学者になります。
以下のコードでなぜ[5 1]と表示されるのかがわからないです。
ご教示いただけると幸いです。

import numpy as np

array2=np.array([[8, 4, 1, 7, 2, 12],[6, 141, 3, 5, 9, 10]])
print(np.argmax(array2,1))

#[5 1]と表示

Answer 1

それがnp.argmaxの仕様です。

>>> array2
array([[  8,   4,   1,   7,   2,  12],
       [  6, 141,   3,   5,   9,  10]])
>>> print(np.argmax(array2,1))
[5 1]
>>> print(np.argmax(array2,axis=1))
[5 1]
>>> print(np.argmax(array2,axis=0))
[0 1 1 0 1 0]

axisが1の時には横方向の最大値となるインデックスを返します。全部で2個あります。
その2個とは、12と141です。12のインデックスは6番目なので5です。141のインデックスは2番目なので1です。それを並べたndarray ［5 1]を返しています。

axisが0の時には縦方向の最大値となるインデックスを返します。全部で6個あります。

英語のマニュアルならnumpy.argmaxを読んでください。
日本語訳はnumpy.argmaxのようですが、英語より分かりにくいかもしれません。

Comments

[hinata0812]

度々申し訳ございません。
print(np.argmax(array2[0:2]))
だと10表示されるのはなぜでしょうか？