[Python] 파이썬 문자열 내 마음대로 정렬하기 - sorted 함수에서 key lambda 사용해서 정렬 기준 여러 개 반영하기

문제

프로그래머스 문자열 내 마음대로 정렬하기

문제 설명
문자열로 구성된 리스트 strings와, 정수 n이 주어졌을 때, 각 문자열의 인덱스 n번째 글자를 기준으로 오름차순 정렬하려 합니다. 예를 들어 strings가 ["sun", "bed", "car"]이고 n이 1이면 각 단어의 인덱스 1의 문자 "u", "e", "a"로 strings를 정렬합니다.

제한 조건
strings는 길이 1 이상, 50이하인 배열입니다.
strings의 원소는 소문자 알파벳으로 이루어져 있습니다.
strings의 원소는 길이 1 이상, 100이하인 문자열입니다.
모든 strings의 원소의 길이는 n보다 큽니다.
인덱스 1의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치합니다.

 

내 소스코드

def solution(strings, n):
    return sorted(strings, key = lambda x: (x[n], x))


strings = ["abce", "abcd", "cdx"]
print(solution(strings, 2))
strings = ["sun", "bed", "car"]
print(solution(strings, 1))

정렬 기준에서 람다를 사용했을 때 다중 조건을 설정할 수 있다는 장점이 있다.

lambda x: (x[n], x) 여기서 첫번째 정렬 기준은 x[n]으로 오름차순 정렬을 하지만, 첫번째 정렬기준에서 같을 때 그다음은 그냥 문자열 x를 오름차순한다.