이전 포스팅에서 이어집니다.
sort함수로 정렬이 된 ndarray의 원본 index를 출력하는 함수입니다.
np.argsort()
의 형식으로 쓰이는데 그 예시를 보여드리겠습니다.
array1 = np.array([2, 4, 1, 3])
array2 = np.argsort(array1)
print(array2)
[2 0 3 1]
이렇게 정렬이 된 array의 원본 인덱스를 반환합니다.
각각 행렬에서 최대값과 최솟값의 인덱스를 반환하는 함수입니다.
array1 = np.array([[2, 4, 1, 3], [5, 8, 9, 11], [14, 4, 21, 9]])
print(np.argmax(array1))
print(np.argmin(array1))
10
2
위의 반환된 값에 따르면 max와 min은 각각 10번째, 2번째에 있다고 나옵니다.
직접 세려보면 최대값인 21과 최솟값인 1이 각각 알맞은 위치에 있다는 것을 알 수 있습니다.
이때 위 두 함수에도 axis 파라미터를 추가해서 원하는 축의 최대값과 최솟값을 알아낼 수 있습니다.
np.argmax(array1, axis=1)
이런 식으로 axis를 설정해주시면 각 row나 col마다의 최대값과 최솟값이 반환이 됩니다.
이것으로 행렬과 관련한 함수의 포스팅을 마치겠습니다.