如何优化计算国家社保号校验位的两个Python字典？

硬编码两个包含36个键值对的字典确实不够高效，不仅代码冗余，后期修改规则也麻烦。这里提供两种更简洁易维护的优化方案：

方案1：用列表+索引映射替代字典 #

利用字母（A-Z）和数字（0-9）的顺序性，把奇偶位置的对应值按顺序存入列表，通过计算字符的索引来获取对应值，避免重复定义键值对：

# 定义奇偶位置的对应值列表：前26个是A-Z的映射，后10个是0-9的映射
# 示例值根据你给出的规则填写，实际替换为你的真实规则
odd_values = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25,
              0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
even_values = [1, 9, 13, 17, 21, 25, 29, 33, 37, 41, 45, 49, 53, 57, 61, 65, 69, 73, 77, 81, 85, 89, 93, 97, 101, 105,
               0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

def get_char_value(char, is_odd_position):
    """根据字符和位置奇偶性获取对应数值"""
    if char.isalpha():
        # 计算字母在列表中的索引（A-Z对应0-25）
        idx = ord(char.upper()) - ord('A')
    elif char.isdigit():
        # 数字对应列表的26-35索引
        idx = 26 + int(char)
    else:
        raise ValueError("仅支持字母或数字字符")
    
    return odd_values[idx] if is_odd_position else even_values[idx]

# 校验位计算示例（假设位置从1开始计数）
def calculate_check_digit(prefix):
    total = 0
    for pos, char in enumerate(prefix, start=1):
        total += get_char_value(char, pos % 2 == 1)
    # 这里替换为你的校验位计算规则，比如取模、减法等
    return str(total % 10)

优点 #

• 代码更简洁，避免了重复的键值对定义
• 维护成本低：修改映射规则只需调整列表中的对应位置值
• 逻辑清晰，新手更容易理解和调试

方案2：用公式计算映射值（如果规则有规律） #

如果奇偶位置的映射值存在数学规律（比如字母的映射值和其ASCII码有线性关系），可以直接用公式计算，连列表都不用定义。比如假设偶数位置字母的映射值满足 (ord(char) - ord('A')) * 4 + 1（对应你给出的A=1, B=9, C=13...），那代码可以简化为：

def get_char_value(char, is_odd_position):
    if char.isalpha():
        char_idx = ord(char.upper()) - ord('A')
        if is_odd_position:
            return char_idx  # 奇数位置A=0, B=1...
        else:
            return char_idx * 4 + 1  # 偶数位置A=1, B=9, C=13...
    elif char.isdigit():
        return int(char)  # 数字奇偶位置值相同的情况
    else:
        raise ValueError("仅支持字母或数字字符")

优点 #

• 代码最精简，完全没有硬编码的映射表
• 规则变更时只需调整公式，无需修改大量数据

你可以根据实际的映射规则选择合适的方案，如果规则没有统一公式，方案1是最稳妥的选择。

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