Lv1_시저암호.py

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