본문 바로가기
Beakjoon&프로그래머스/파이썬

[프로그래머스/파이썬] 문자열 내 마음대로 정렬하기

by 현장 2021. 12. 19.

-코드

처음에는 이렇게 짰습니다.

def solution(strings, n):
        answer = strings.sort(key=lambda x: x[n])
    return answer

하지만 이렇게 짜면 출력 값이 none이 나오게 되어서 sorted를 이용해서 짜보았습니다.

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

이렇게 짜니 출력이 1개 맞고 1개 틀려서 이유를 보니 정렬에 문제가 더 있어서 sorted를 다음 코드 같이 한번 더 사용하여 해결을 하였습니다.

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