2학년 2학기/데이터 사이언스 입문
Sorting
kkkkk1023
2024. 10. 5. 17:32
sort: 요소를 오름차순으로 정렬
argsort: 요소를 오름차순으로 정렬했을 때 각각의 원본의 요소 인덱스를 출력
축(Axis) 기준으로 정렬
axis = 0: 열마다 정렬 시킨다.
axis = 1: 행마다 정렬 시킨다.
부분 정렬
np.partition(배열, num): [0 ~ num-1]는 [num ~ last]보다 확실하게 작은 수로 구성되어 있음을 보장한다. 단, 각 구간 내의 정렬은 보장하지 않는다.
사용예시 - 나는 Top3를 뽑고 싶어 순서는 상관없어!라고 했을 때 부분 정렬 3을 사용하면 상위 3을 바로 추려낼 수 있다.
축 기준으로도 진행할 수 있다.
axis = 1인 경우 행을 기준으로 행 마다 가장 작은 2개가 앞으로 오게 되고 나머지는 뒤로 밀려난다.
axis = 0인 경우 열을 기준으로 열마다 가장 작은 2개가 위로 오게되고 나머지는 아래로 밀려난다.