python numpyの配列内の最大値のインデックス番号を取得する

pip install numpy

# Successfully installed numpy-1.19.4

import numpy as np

arr = np.array([2, 3, 1, 4, 5, 6])

print(np.argmax(arr))
# 5

# 同じ値の場合は、最初の方のインデックス番号を返す
arr = np.array([2, 6, 1, 4, 5, 6])

print(np.argmax(arr))
# 1

import numpy as np

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

# フラット化された値が返ってきます
print(np.argmax(arr))
# 8

# unravel_indexを使用すると8ではなく(2,2)で取得することが可能です
print(np.unravel_index(np.argmax(arr, axis=None), arr.shape))
# (2, 2)