이상한 문자 만들기 / 프로그래머스, 파이썬 코딩테스트 연습 연습문제

문제

문자열 s는 한 개 이상의 단어로 구성되어 있습니다. 각 단어는 하나 이상의 공백문자로 구분되어 있습니다. 각 단어의 짝수번째 알파벳은 대문자로, 홀수번째 알파벳은 소문자로 바꾼 문자열을 리턴하는 함수, solution을 완성하세요.

 

 

제출답안

def solution(s):
    ls_char = [[word[i:i+1].lower() if i%2 else word[i:i+1].upper()for i in range(len(word))] for word in s.split(' ')]
    ls_word = [''.join(chars) for chars in ls_char]
    answer = ' '.join(ls_word)
    return answer

 

 

제출 후 개선답안

len(range(x))를 enumerate로 간소화 가능
두 번 리스트화 하는 과정을 map 함수와 lambda를 이용해서 축약가능

def solution(s):
    answer = ' '.join(map(lambda x: ''.join([chr.lower() if i % 2 else chr.upper() for i, chr in enumerate(x)]), s.split(" ")))
    
    return answer

 

 

개선 전 코드 결과 / 개선 후 코드 결과

문제 바로가기