[PYTHON/NUMPY] ndarray 클래스 : argmax 함수에서 axis 인자를 사용해 각 축별 최대 요소값 인덱스 배열 구하기

■ ndarray 클래스에 대해 argmax 함수에서 axis 인자를 사용해 각 축별 최대 요소값 인덱스 배열을 구하는 방법을 보여준다.

 

▶ 예제 코드 (PY)

import numpy as np

sourceList = [(1, 2, 3), (4, 5, 6)]

sourceNDArray = np.array(sourceList)

targetNDArray = np.argmax(sourceNDArray, axis = 0) # 또는 axis = 1

print(targetNDArray)

"""
[1 1 1]
"""