리스트에서 상위 n개 데이터의 인덱스, 값 반환하기
by Cori리스트에서 상위 n개 데이터의 위치, 값을 반환하는 방법에 대해 살펴본다.
코드
def get_top_n(list_a, num):
'''
list_a: [int, int, ...]
num: int, 추출하고 싶은 개수
'''
tmp = list_a.copy() # list의 값 복사
tmp.sort() # 복사한 리스트 오름차순 정렬
top_num = tmp[-num:] # 지정한 개수만큼 뒤에서부터 추출
top_idx = [list_a.index(x) for x in top_num] # 해당 값 위치 추출
return top_num, top_idx
'CS > Python' 카테고리의 다른 글
[자료구조] 깊이 우선 탐색(DFS)과 너비 우선 탐색(BFS) (0) | 2021.09.14 |
---|---|
2차원 리스트 (0) | 2021.09.09 |
[자료구조] 완전탐색/이분탐색 (0) | 2021.09.03 |
ASCII 코드 연산 - chr, ord (0) | 2021.08.28 |
[자료구조] 스택 / 큐 (0) | 2021.08.26 |
블로그의 정보
코딩하는 오리
Cori