[PYTHON] 2차원 배열 정렬하기

2차원 배열 정렬하기

파이썬 sort() 내장함수로 1차원 배열을 쉽게 정렬할 수 있다. 

arr = [2,3,4,5,1]

arr.sort() 				 
# 오름차순

arr.sort(reverse = True) 
# 내림차순

 

2차원 배열의 정렬은 기준을 명시해주는 것이 필요한데

# 기준 명시X (-> arr[i]를 기준으로 오름차순 정렬)
arr = [[2,3],[1,2],[0,4]]

arr.sort()

print(arr)
# [[0, 4], [1, 2], [2, 3]]

 

기준을 명시한다면

arr = [[2,3],[1,2],[0,4]]

arr.sort(key=lambda x:x[0])

print(arr)
# [[0, 4], [1, 2], [2, 3]]


# 이 때, x값에 -를 취하면 내리림차순 정렬 가능
# arr.sort(key=lambda x: -x[0])

 

만약, 다중 조건을 명시하고 싶다면

arr = [[2, 3], [1, 2], [0, 4], [2, 2]]

arr.sort(key=lambda x: (x[1], x[0]))

print(arr)
# [[1, 2], [2, 2], [2, 3], [0, 4]]

 

로 할 수 있다!