[프로그래머스/파이썬] 시저 암호

-코드

def solution(s, n):
    answer = ''
    for i in s:
        if i != ' ':
            if i.isupper():
                answer += chr((ord(i) - ord('A') + n) % 26 + ord('A'))
            elif i.islower():
                answer += chr((ord(i) - ord('a') + n) % 26 + ord('a'))
        else:
            answer += ' '
    return answer