백준 11651) 좌표 정렬하기 2 [ python , lambda ]

 

분류 : 정렬

 

keypoint : python lambda, sort(key= )

 

code

# n 번의 숫자입력을 받습니다.
n = int(input()) 

# 좌표를 저장할 리스트
coor = []
for i in range(0,n):
    a,b = input().split(' ') 
    coor.append([int(a),int(b)])
    
    #리스트에 n 개의 좌표를 입력받는데
    #coor.append(list(map(int,input().split(' ')) 으로 대체할 수 있습니다.

#lambda 의 개념을 알아야 하는데 이건 아래에서 따로 다루겠습니다. 
coor.sort(key=lambda x: (x[1],x[0]))

for coord in coor:
    print(coord[0], coord[1])

---

lambda 

람다는 파이썬에서 중요하게 사용되는 함수입니다.

lambda 인자 : 표현식

 

람다는 다른 함수와 달리 def 로 선언하지 않고 인라인으로 선언하고 사용하여 사용할 수 있습니다.

예를들어 

def add(x, y):
	return x+y

 

라는 함수가 있다면 람다로 아래와 같이 표현할 수 있습니다.

add = lambda x, y: x+y

 

보통은 map 이나 sorted 와 같이 사용되게 되는데 이번 문제에서 활용한 것 처럼 sort 와 함께 사용해보겠습니다. 

 

 sort() 함수는 해당 리스트를 정렬하는데 사용됩니다. 이때 key 라는 인자를 활용 할 수 있는데 list 에서 key 를 사용하면 어떤 값을 기준으로 정렬할 것인지를 나타낼 수 있습니다.

 

my_list = ['apple', 'banana', 'cherry']

my_list.sort(key= lambda x: len(x))

 

라고 한다면  my_list 는 

['apple', 'cherry', 'banana']

 

로 저장됩니다. 이 처럼 key 를 사용하면 어떤 값을 통해 정렬할 것인지를 결정할 수 있다. 

 

우리 문제에서는

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

 

로 key 에 lambda 함수를 사용했는데 lambda x : x[1],x[0] 은 

def _key_function(x):
	return (x[1],x[0])

 

과 같다. 

 

즉 x[1] 과 x[0] 이 정렬 기준이라는 것인데, 이때 x[1] 기준으로 오름차순 정렬을 하고 만약 x[1] 값이 같다면 x[0] 을 기준으로 오름차순 정렬을 하게 된다.

 

이건 그냥 외우자!