파이썬 numpy.argmin 함수 활용하기

Numpy Argmin 함수 소개:

Numpy의 `numpy.argmin` 함수는 배열에서 최솟값의 인덱스를 반환하는 함수입니다. 이 함수는 주어진 배열에서 가장 작은 값을 가지는 요소의 인덱스를 찾아줍니다. 데이터의 최솟값 위치를 찾는 데에 유용하게 활용됩니다.

기본 사용법:

`numpy.argmin` 함수의 기본 사용법은 다음과 같습니다.

import numpy as np

# 1차원 배열에서 최솟값의 인덱스 찾기
array = np.array([3, 1, 4, 2, 5])
argmin_index = np.argmin(array)
print("최솟값의 인덱스:", argmin_index)

위 예시 코드에서 `np.argmin(array)`는 주어진 1차원 배열에서 최솟값을 가지는 요소의 인덱스를 찾습니다.

예시 코드:

• 다차원 배열에서 최솟값의 인덱스 찾기:

# 다차원 배열에서 최솟값의 인덱스 찾기
array_2d = np.array([[3, 1, 4], [2, 5, 0]])
array_3d = np.array([[[3, 1], [4, 2]], [[5, 0], [2, 4]]])

argmin_2d = np.argmin(array_2d)  # 모든 원소 중 최솟값의 인덱스
argmin_2d_axis_0 = np.argmin(array_2d, axis=0)  # 각 열별 최솟값의 인덱스
argmin_2d_axis_1 = np.argmin(array_2d, axis=1)  # 각 행별 최솟값의 인덱스

argmin_3d = np.argmin(array_3d)  # 모든 원소 중 최솟값의 인덱스
argmin_3d_axis_0 = np.argmin(array_3d, axis=0)  # 각 깊이별 최솟값의 인덱스
argmin_3d_axis_1 = np.argmin(array_3d, axis=1)  # 각 행별 최솟값의 인덱스
argmin_3d_axis_2 = np.argmin(array_3d, axis=2)  # 각 열별 최솟값의 인덱스

print("2차원 배열 최솟값의 인덱스:", argmin_2d)
print("2차원 배열 각 열별 최솟값의 인덱스:", argmin_2d_axis_0)
print("2차원 배열 각 행별 최솟값의 인덱스:", argmin_2d_axis_1)

print("3차원 배열 최솟값의 인덱스:", argmin_3d)
print("3차원 배열 각 깊이별 최솟값의 인덱스:", argmin_3d_axis_0)
print("3차원 배열 각 행별 최솟값의 인덱스:", argmin_3d_axis_1)
print("3차원 배열 각 열별 최솟값의 인덱스:", argmin_3d_axis_2)

결론:

`numpy.argmin` 함수는 다차원 배열에서 최솟값을 가지는 요소의 인덱스를 찾는 유용한 함수입니다. 데이터의 최솟값 위치를 파악하는 데에 활용되며, 다양한 축(axis)을 지정하여 배열의 최솟값의 인덱스를 계산할 수 있습니다. 예시 코드를 통해 `numpy.argmin` 함수의 활용법을 익혀보세요.