파이썬에서 튜플로 구성된 리스트 정렬할 때

2022. 2. 20. 20:28파이썬/알고리즘

위는 예시로 주어진 문제이다.

import sys

input = sys.stdin.readline

arr = [tuple(map(int,input().split())) for _ in range(int(input()))]
arr.sort(key= lambda x : (x[0],x[1]))
for i in arr:
    print(i[0],i[1])

key에다 람다함수로 iterable한 형식으로 넣어주면 순서대로 오름차순으로 정렬해준다.

위로 예시로 들면 (x[0], x[1]) 이렇게 튜플 형식(리스트여도 된다.) x[0]을 기준으로 정렬하되 같을 때는 x[1]로 정렬한다.

내림차순으로 정렬을 하고 싶다면 -x[0] 같은 형식으로 음수 부호를 붙여주면 된다.

 

반응형