여러개의 List를 기준으로 정렬하기

지난포스팅에서 다수의 List가 있을 때 특정 List를 기준으로 정렬하는 방법에 대해서 포스팅 하였습니다.

 

하지만 실제 사용할 때는 특정 List뿐만 아니라, 다수의 List를 기준으로 Table을 정리하는 경우가 있습니다.

 

Excel에서 사용 가능한 다수 List를 기준으로한 정렬

이 경우에, 정렬할 list의 우선위를 정해주고 정렬을 하게 됩니다.

 

예를들어서 List1이 [a,a,a,b,b,e,d,s,a], List2가 [1,6,7,9,7,5,2,1,3]라고 한다면

 

List1의 1순위, List2를 2순위로 설정할 경우 list가 a인 값에 list2는 1,6,7,3이 존재하게 됩니다.

 

그럼 이제 list1의 값이 같은 List를 부분배열로 하여 한번더 정렬을 시켜주면 됩니다.

 

이제 이 정렬할 List의 우선순위가 N개 있다고 치면

1. N-1번째 List에서 k, k+1 Index인 Element의 값이 다른 경우를 찾는다.

2. 0~k번째 까지의 list를 부분배열로 선정한다.

3. 부분배열을 N번째 List를 기준으로 정렬

4. N+1번째에서 반복

 

이때 N-1번째에서 같은값은 전체에서 찾는다면, 이전의 정렬이 망가지기 때문에 주의해야 합니다 :)

 

 Python으로 패키지없이 구현된 여러개의 List기준으로 정렬하기 소스코드입니다.

#bubble sort로 list를 정렬하며, 이때 list의 n번째 Index의 값을 기준으로 정렬
def bubble_sort(DT_list, n=None):
    for i in range(len(DT_list)):
        for j in range(len(DT_list)-i-1):
            if n !=None:
                if DT_list[j][n] > DT_list[j+1][n]:
                    #Normal bubble Sort와 다르게 해당 값을 움직이는게 아니라
                    #해당 Element의 n번째Index의 값을 기준으로 판단 후
                    #해당 Element Set을 통째로 교환
                    temp = DT_list[j]
                    DT_list[j] = DT_list[j+1]
                    DT_list[j+1] = temp
    return DT_list


def multi_sort(DT_list, sort_order):    
    #가장 첫번째 정렬 우선순위 순서에서는 일반적인 Sort를 진행
    DT_list = bubble_sort(DT_list, sort_order[0])
    #다음 정렬 우선순위에서는 부분배열을 생성 후 부분 정렬을 진행
    for i in range(len(sort_order[:-1])):
        #k는 부분배열의 시작점을 저장합니다.
        k=0             
        #List의 N-1번째 정렬기준 List를 확인하면서, k와 k+1번째 Element가 다른 지점을 확인합니다.
        for j in range(len(DT_list)-1):                 
            #마지막 Index에 추가로 None을 추가하는 방법도 있지만, 이번에는 해당 방법이 아니라
            #이때 k+1번째 Index가 마지막 Index인 경우와, 아닌 경우를 나눠서 판정을 진행합니다.           
            if DT_list[j][sort_order[i]]!=DT_list[j+1][sort_order[i]] or j+1 == len(DT_list)-1:
                #일반적인 경우 index k~j 사이의 배열을 부분 배열로 Extract하고
                #k+1번째 Index가 마지막 Index인 경우 k~j+1의 Index를 기준으로 부분배열을 Extract합니다.
                temp_dt = DT_list[k:j+1] if j+1 != len(DT_list)-1 else DT_list[k:j+2]             
                #부분배열을 Sorting한 후 원본 배열에 돌려줍니다.
                if j+1 == len(DT_list)-1:
                    DT_list[k:j+2] = bubble_sort(temp_dt, sort_order[i+1])
                else:                
                    DT_list[k:j+1] = bubble_sort(temp_dt, sort_order[i+1])
                #배열 정렬 완료후 k를 갱신해줍니다.
                k = j+1
    return DT_list
            


import random
#Random한 25개의 값을 갖는 4개의 List를 듭니다.
test1 = [random.randint(0,3) for i in range(25)]
test2 = [random.randint(0,3) for i in range(25)]
test3 = [random.randint(0,3) for i in range(25)]
test4 = [random.randint(0,3) for i in range(25)]
#위 4개의 경우 0~24의 Index를 갖고, test1~test4 까지의 col을 갖는
#DataFrame과 동일한 형태라고 할 수 있습니다.
#Operation을 위해서 zip함수로 4개의 List를 묵어줍니다.
#묶어주면 ((test1[0],test2[0],test3[0],test4[0]),(test1[1],test2[1],test3[1],test4[1])...)
#이제 0, 3, 2 ,1번째 List를 우선순위로 정렬을 진행합니다.
test_result = multi_sort((list(zip(test1, test2, test3, test4))),[0,3,2,1])
print(test_result)
[(0, 3, 1, 0),
 (0, 0, 2, 1),
 (0, 1, 2, 0),
 (0, 0, 3, 2),
 (0, 1, 3, 1),
 (0, 1, 3, 1),
 (0, 0, 2, 3),
 (0, 2, 3, 3),
 (1, 2, 0, 0),
 (1, 1, 2, 0),
 (1, 1, 0, 1),
 (1, 1, 2, 1),
 (1, 1, 3, 1),
 (1, 1, 3, 1),
 (1, 2, 3, 1),
 (1, 2, 1, 2),
 (1, 0, 3, 3),
 (2, 0, 0, 0),
 (2, 1, 0, 0),
 (3, 1, 0, 0),
 (2, 3, 3, 0),
 (2, 3, 2, 1),
 (2, 3, 2, 1),
 (2, 0, 3, 1),
 (2, 0, 3, 2)]