PTA Python 7-3 字符串加密

用户在一行中输入一个包括大小写字母和数字的字符串，编程将其中的大写字母用该字母后的第5个字母替代，小写字母用该字母后的第3个字符替代，其他字符原样输出，实现字符串加密。如：A--F,b--e,Z---E,w--z

输入格式:

输入一个至少包含一个字母的字符串

输出格式:

加密后的字符串

输入样例:

在这里给出一组输入。例如：

Life is short, you need Python!

输出样例:

在这里给出相应的输出。例如：

Qlih lv vkruw, brx qhhg Ubwkrq!

答案:

 

def encrypt(str_is):
    str_end = ""
    for char in str_is:
        if char.isupper():
            str_end += chr((ord(char) - ord('A') + 5) % 26 + ord('A'))
        elif char.islower():
            str_end += chr((ord(char) - ord('a') + 3) % 26 + ord('a'))
        else:
            str_end += char
    return str_end


str_input = input()
str_encrypt = encrypt(str_input)
print(str_encrypt)