[프로그래머스] 시저 암호

나의 코드 

def solution(s, n):
    s = list(s)
    for i in range(len(s)):
        if s[i].isupper():
            s[i] = chr((ord(s[i]) - ord('A') + n) % 26 + ord('A'))
        elif s[i].islower():
            s[i] = chr((ord(s[i]) - ord('a') + n) % 26 + ord('a'))
    return "".join(s)

 

따로 공백을 처리 해주려고 했는데 list의 isupper, islower 함수를 사용하면 편하다.