■ 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 = 1) print(sourceNDArray[..., targetNDArray][..., 0]) """ [3 6] """ |