문제는 간단하지만 단어의 길이가 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])