如何使用Python实现字母表的字母位移功能

原有代码存在的问题 #

• 仅支持单个小写字母输入，无法匹配需求中处理完整单词、大小写字母的场景
• 硬编码26个字母的映射字典，扩展性差，调整位移量需要修改全部映射关系，维护成本高

优化实现思路 #

利用ASCII码运算替代硬编码映射：

1. 小写字母ASCII范围为97(a)~122(z)，大写字母为65(A)~90(Z)
2. 字符减去对应大小写的基准ASCII值得到0-25的序号，加上位移量后对26取模，即可处理z→a、Z→A的边界情况
3. 遍历输入字符串的每个字符单独处理，拼接后得到最终结果

优化后代码 #

def shift_string(input_str, offset=1):
    result = []
    for char in input_str:
        if char.islower():
            base = ord('a')
            shifted = (ord(char) - base + offset) % 26
            result.append(chr(base + shifted))
        elif char.isupper():
            base = ord('A')
            shifted = (ord(char) - base + offset) % 26
            result.append(chr(base + shifted))
        else:
            # 非字母字符默认保留，如有需要可替换为抛出异常逻辑
            result.append(char)
    return ''.join(result)

测试示例 #

print(shift_string("Car")) # 输出：Dbs
print(shift_string("zZ")) # 输出：aA
print(shift_string("Hello World!", 2)) # 输出：Jgnnq Yqtnf!

优化优势 #

• 代码更简洁，无需维护大量硬编码映射，内存占用更低
• 支持自定义位移量，默认1位可满足当前需求，也可灵活调整实现不同位移规则的凯撒密码
• 自动适配大小写字母，支持完整字符串直接输入，符合使用场景要求
• 非字母字符的处理逻辑可自定义，适配不同的输入校验需求

内容的提问来源于stack exchange，提问作者Sneezy_HD