JadenCase 문자열 만들기 [Level 2] Python

문제 설명

JadenCase란 모든 단어의 첫 문자가 대문자이고, 그 외의 알파벳은 소문자인 문자열입니다. 단, 첫 문자가 알파벳이 아닐 때에는 이어지는 알파벳은 소문자로 쓰면 됩니다. (첫 번째 입출력 예 참고)
문자열 s가 주어졌을 때, s를 JadenCase로 바꾼 문자열을 리턴하는 함수, solution을 완성해주세요.

제한 조건

• s는 길이 1 이상 200 이하인 문자열입니다.
• s는 알파벳과 숫자, 공백문자(" ")로 이루어져 있습니다.
  - 숫자는 단어의 첫 문자로만 나옵니다.
  - 숫자로만 이루어진 단어는 없습니다.
  - 공백문자가 연속해서 나올 수 있습니다.

입출력 예

s(string): "3people unFollowed me"
Return: "3people Unfollowed Me"

s(string): "for the last week"
Return: "For The Last Week"

s(string): "he(공백 2개)llo"
Return: "He(공백 2개)Llo"

코드

import re
def solution(s):
    answer = list(''.join(map(lambda x: x.capitalize(), s.split()))) # 공백 단위로 문자열을 나누고 첫글자를 대문자로 변환한 뒤, 리스트 구성
    
    for j in [i.start() for i in re.finditer(' ', s)]: # 공백 문자 시작 인덱스 찾기
        answer.insert(j, ' ') # 리스트에 j 공백 시작 인덱스만큼 공백 넣기
    return ''.join(answer)

string.capitalize(): 영문 문자열의 첫 글자를 대문자로 변환하여 반환해줌
re.finditer(패턴, 문자열, 플래그): findall()과 유사하지만 패턴에 맞는 문자열의 리스트가 아닌 iterator 형식으로 반환
-> iterator 형식.start(): 패턴 문자열의 시작