Numpy min, max 값과 위치를 구하는 법(min,max,argmin,argmax)

안녕하세요 jay입니다.
오늘은 Numpy array에서 min, max,median 값을 구하는 법에 대해 알아보도록 하겠습니다.


# Min, Max, Argmin, Argmax

import numpy as np

arr = np.random.randint(1,10,(4,3))
print(arr)

"""
[[2 5 6]
 [9 9 2]
 [7 2 1]
 [9 8 1]]
"""

# max : arr에서 가장 큰 값을 출력
print(np.max(arr)) # 9
print(arr.max()) # 9

# min : arr에서 가장 작은 값을 출력
print(np.min(arr)) # 1
print(arr.min()) # 1

# argmin : 최소의 위치를 반환
print(np.argmin(arr)) # 8 (인덱스 기준이므로 8번째)
print(arr.argmin()) # 8 (인덱스 기준이므로 8번째)

# argmax : 최대의 위치를 반환
print(np.argmax(arr)) # 3 (인덱스 기준이므로 3번째)
print(arr.argmax()) # 3 (인덱스 기준이므로 3번째)

위와 같이 전체에서의 min, max을 구할 수 있습니다. 슬라이싱을 통해서 원하는 행과, 열의 max, min의 값을 구할 수 있습니다.

댓글