문제는 간단하지만 단어의 길이가 1인 경우를 생각하지 못해 런타임 에러가 나기도 했다. 다양한 조건을 확인하고 문제에 접근하자.
def solution(s):
    s = s.split(' ')
    answer = []
    for word in s:
        word = word.upper()
        if len(word) > 1: word = word[0] + word[1:].lower()
        answer.append(word)
    return ' '.join([x for x in answer])