■ ndarray 클래스에 대해 argmax 함수에서 axis 인자를 사용해 각 축별 최대 요소값 인덱스 배열을 구하는 방법을 보여준다.
▶ 예제 코드 (PY)
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
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] """ |