[test]문자열 내 마음대로 정렬하기-파이썬

Joy·2020년 4월 19일
1

Coding Test

목록 보기
19/48

문제 설명
문자열로 구성된 리스트 strings와, 정수 n이 주어졌을 때, 각 문자열의 인덱스 n번째 글자를 기준으로 오름차순 정렬하려 합니다. 예를 들어 strings가 [sun, bed, car]이고 n이 1이면 각 단어의 인덱스 1의 문자 u, e, a로 strings를 정렬합니다.
제한 조건
strings는 길이 1 이상, 50이하인 배열입니다.
strings의 원소는 소문자 알파벳으로 이루어져 있습니다.
strings의 원소는 길이 1 이상, 100이하인 문자열입니다.
모든 strings의 원소의 길이는 n보다 큽니다.
인덱스 1의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치합니다.

입출력 예
strings	n	return
[sun, bed, car]	1	[car, bed, sun]
[abce, abcd, cdx]	2	[abcd, abce, cdx]
입출력 예 설명
입출력 예 1
sun, bed, car의 1번째 인덱스 값은 각각 u, e, a 입니다. 이를 기준으로 strings를 정렬하면 [car, bed, sun] 입니다.
입출력 예 2
abce와 abcd, cdx의 2번째 인덱스 값은 c, c, x입니다. 따라서 정렬 후에는 cdx가 가장 뒤에 위치합니다. abce와 abcd는 사전순으로 정렬하면 abcd가 우선하므로, 답은 [abcd, abce, cdx] 입니다.

내 풀이

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

sorted를 이용해서 바로 리턴하기,
key로 기준 정해주기. - strings요소 들의 n 인덱스 (lamda함수 사용해서 한줄로 써주기)

결과: 테스트 2를 통과하지 못함. 인덱스 값이 같은 두 수를 샂ㄴ순으로 정리해줘야하기 때문
보완: 미리 주어진 strings를 sorted한 후 내 기준에 따라 다시 sorted

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

참고 자료

https://docs.python.org/ko/3/howto/sorting.html

키 함수
list.sort()와 sorted()는 모두 비교하기 전에 각 리스트 요소에 대해 호출할 함수를 지정하는 key 매개 변수를 가지고 있습니다.
key 매개 변수의 값은 단일 인자를 취하고 정렬 목적으로 사용할 키를 반환하는 함수여야 합니다

키를 기준으로 정렬할 때, lambda 함수를 사용하여 새롭게 정의한 키(custom key function, logic)를 사용할 수도 있습

예를 들어:

>>>
student_tuples = [
    ('john', 'A', 15),
    ('jane', 'B', 12),
    ('dave', 'B', 10),
]
sorted(student_tuples, key=lambda student: student[2])   # sort by age
[('dave', 'B', 10), ('jane', 'B', 12), ('john', 'A', 15)]

ps.

람다표현식을 쓰긴썼는데 어쨔서 작동는지는 아직 이해가 안간다

profile
roundy

0개의 댓글