NumPy는 Python에서 사용되는 강력한 수학 및 과학 연산 라이브러리입니다. 이를 통해 다차원 배열을 효율적으로 처리할 수 있습니다. 2차원 배열에서 특정 열을 정렬하려면 다음과 같은 단계를 따를 수 있습니다.

NumPy를 설치

NumPy가 설치되어 있지 않은 경우, 다음과 같은 명령을 사용하여 설치할 수 있습니다.

pip install numpy

NumPy 배열 생성

먼저, NumPy 배열을 생성해야 합니다. 이를 위해 numpy.array 함수를 사용하면 됩니다. 예를 들어, 다음과 같은 2차원 배열을 생성해보겠습니다.

import numpy as np

# 2차원 배열 생성
arr = np.array([[4, 2, 9],
                [3, 5, 1],
                [7, 6, 8]])

특정 열을 기준으로 정렬

numpy.argsort 함수를 사용하여 배열의 특정 열을 기준으로 정렬할 수 있습니다. 이 함수는 원본 배열을 변경하지 않고, 정렬된 인덱스를 반환합니다.

# 열을 기준으로 정렬된 인덱스 반환
sorted_indices = np.argsort(arr[:, 열의 인덱스])

# 정렬된 인덱스를 사용하여 배열 정렬
sorted_arr = arr[sorted_indices]

위 코드에서 열의 인덱스는 정렬하려는 열의 인덱스 값입니다. 예를 들어, 첫 번째 열을 기준으로 정렬하려면 0을 사용하면 됩니다.

정렬된 배열을 확인하기 위해 다음과 같은 코드를 사용할 수 있습니다.

# 정렬된 배열 출력
print(sorted_arr)

이제 특정 열을 기준으로 정렬하는 방법을 알았습니다. 이를 적용하여 다양한 2차원 배열에 대해 원하는 열을 정렬할 수 있습니다.