본문 바로가기
programming language/Python

Python 다차원 리스트 정렬 방법 itemgetter

by __observer__ 2018. 3. 27.
반응형

저는 증권사 API를 사용하여 매일 기관 외국인 10억이상 순매수 종목에 대해서 검색을 수행하고 있습니다. 프로그래밍 언어로 Python 을 사용하고 있고, API 로 부터 추출된 종목 리스트를 전일비 상하락율에 따라 정렬을 해서 포스팅을 합니다.

이에 대한 예제는 아래와 같습니다. 예로 든 category 데이터에서 3 번째 인덱스 값이 전일 대비 상하락율입니다. 해당 인덱스에 따라서 전체 데이터가 정렬 되도록 하기 위해서 operator 모듈의 itemgetter를 사용합니다. itemgetter 에 3 이라는 정렬의 기준이 되는 인덱스 값을 넣어 줍니다. 기본 정열은 오름차순입니다.




내림차순으로 정열을 하고 싶은 경우에는 reverse 옵션을 True 로 설정하면 됩니다.

3 번째 인덱스가 첫 번째 정렬 조건이고 4 번째 인덱스가 두 번째 정렬 조건이라고 한다면 itemgetter(3, 4) 와 같이 순서대로 기준이 되는 인덱스를 넣어 주면 됩니다.

위 코드를 실행해 보면 아래와 같은 결과를 확인할 수 있습니다.



반응형

댓글