JadenCase 문자열 만들기

• 문제 설명
  JadenCase란 모든 단어의 첫 문자가 대문자이고, 그 외의 알파벳은 소문자인 문자열입니다. 문자열 s가 주어졌을 때, s를 JadenCase로 바꾼 문자열을 리턴하는 함수, solution을 완성해주세요.

• 제한 조건
  s는 길이 1 이상인 문자열입니다.
  s는 알파벳과 공백문자(" ")로 이루어져 있습니다.
  첫 문자가 영문이 아닐때에는 이어지는 영문은 소문자로 씁니다. ( 첫번째 입출력 예 참고 )

def solution(s):
    answer = ''
    # 문자열을 리스트로 만들기
    arr = list(s)
    count = 0
    # 리스트 반복문 실행
    for i in range(0,len(arr)):
    	# 문자열 중에 대문자를 소문자로 바꾸기
        arr[i] = arr[i].lower()
        #만약 공백이면 그 다음 문자가 단어의 첫글자이므로 count 초기화해줌
        if arr[i] == " ":
            count = 0
        #공백이 아니고 count가 0이면 첫글자이면 대문자로 바꾸기
        if count == 0 and arr[i] != " ":
            arr[i] = arr[i].upper()
            # 그 다음 문자는 첫글자가 아니니 count 1로 할당
            count = 1
    #리스트를 다시 문자열로 바꿔주는 join 메서드 사용
    answer = "".join(arr)
    # 값 반환
    return answer

문제 출처: https://programmers.co.kr/learn/courses/30/lessons/12951