소소한 컴퓨터 이야기

리스트에서 상위 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

이렇게 줄여써도 ok

'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

활동하기