Python numpy에서 numpy.argsort()
를 이용하여 배열을 정렬할 수 있습니다.
다양한 예제와 함께 알아보겠습니다.
1. 오름차순으로 배열 정렬
numpy.argsort(arr)
은 배열 arr을 정리하고 정렬된 Index 배열을 리턴합니다.
- Index 배열은, 정렬된 배열의 값이 실제 배열에서 위치하고 있는 Index를 의미
arr[sorted_index]
는 오름차순으로 정렬된 배열 리턴
import numpy as np
arr = np.array([3, 1, 4, 2, 5])
sorted_index = np.argsort(arr)
sorted_arr = arr[sorted_index]
print("arr:", sorted_arr)
print("index:", sorted_index)
위 코드 실행 결과를 보면, 정렬된 배열의 0번 Index는 1이고, 1은 실제 배열에서 Index 1에 위치해있습니다. 따라서 Index 배열에서 Index 0은 1의 값을 갖게 됩니다.
arr: [1 2 3 4 5]
index: [1 3 0 2 4]
2. 내림차순으로 배열 정렬
오름차순으로 정렬된 배열에 [::-1]
를 사용하여 순서를 뒤집으면 내림차순으로 정렬된 배열을 얻을 수 있습니다.
import numpy as np
arr = np.array([3, 1, 4, 2, 5])
sorted_index = np.argsort(arr)
sorted_arr = arr[sorted_index]
reversed_arr = sorted_arr[::-1]
print("reversed arr:", reversed_arr)
Output:
reversed arr: [5 4 3 2 1]
3. dtype 데이터 정렬
dtype
으로 정의된 객체의 배열도 argsort()
로 정렬할 수 있습니다.
아래 예제에서 name-age
쌍을 dtype으로 정의하였습니다.
4개 데이터가 있는 배열을 argsort()
로 정렬하는데, order=('age', 'name')
은 age를 우선으로 정렬하고, age가 동일하면 name으로 정렬하라는 의미입니다.
import numpy as np
dt = np.dtype([('name', 'S10'), ('age', int)])
arr = np.array([('Alice', 25), ('Bob', 30), ('Charlie', 30), ('David', 35)], dtype=dt)
sorted_index = np.argsort(arr, order=('age', 'name'))
sorted_arr = arr[sorted_index]
print(sorted_arr)
Output:
[(b'Alice', 25) (b'Bob', 30) (b'Charlie', 30) (b'David', 35)]
Loading script...
Related Posts
- Python - Yaml 파일 파싱하는 방법
- Python - 파일 내용 삭제
- Python - for문에서 리스트 순회 중 요소 값 제거
- Python - 두 리스트에서 공통 요소 값 찾기
- Python - 문자열 앞(뒤)에 0으로 채우기
- Python - 공백으로 문자열 분리
- Python - 중첩 리스트 평탄화(1차원 리스트 변환)
- Python - 16진수 문자열을 Int로 변환
- Python - 두 날짜, 시간 비교
- Python f-string으로 변수 이름, 값 쉽게 출력 (변수명 = )
- Python - nonlocal과 global 사용 방법
- Python 바다코끼리 연산자 := 알아보기
- Python - pip와 requirements.txt로 패키지 관리
- Python - 딕셔너리 보기 좋게 출력 (pprint)
- Python - Requests 사용 방법 (GET/POST/PUT/PATCH/DELETE)
- Python - 온라인 컴파일러 사이트 추천
- Python - os.walk()를 사용하여 디렉토리, 파일 탐색
- Python - 문자열 비교 방법
- Python - Text 파일 읽고 쓰는 방법 (read, write, append)
- Python - 리스트에서 첫번째, 마지막 요소 가져오는 방법
- Python - 두개의 리스트 하나로 합치기
- Python - 리스트의 마지막 요소 제거
- Python - 리스트의 첫번째 요소 제거
- Python 소수점 버림, 4가지 방법
- Python 코드 안에서 버전 확인 방법
- Python 소수점 반올림, round() 예제
- Python - 리스트 평균 구하기, 3가지 방법
- Python - bytes를 String으로 변환하는 방법
- Python - String을 bytes로 변환하는 방법
- Python 버전 확인 방법 (터미널, cmd 명령어)
- Python - 람다(Lambda) 함수 사용 방법
- Python - dict 정렬 (Key, Value로 sorting)
- Python - range() 사용 방법 및 예제
- Python - 리스트를 문자열로 변환
- Python - 문자를 숫자로 변환 (String to Integer, Float)