如何循环Python中if语句的elif分支以优化重复代码？

嘿，作为Python新手遇到这种冗余代码的问题太正常啦！我来帮你把那些冗长的elif分支改成更简洁的实现，不管是用循环还是更高效的字典映射都能搞定～

先看你原来的代码问题（模拟示例） #

你之前的代码大概是这样的对吧？一堆重复的判断逻辑：

# 假设chars是你要判断的字符集合，比如chars = ['a', 'b', 'c', ...]
if code == ord(chars[0]):
    return chars[0]
elif code == ord(chars[1]):
    return chars[1]
elif code == ord(chars[2]):
    return chars[2]
# ... 还有N多类似的elif分支

这种写法不仅冗余，后期维护也麻烦，确实该优化！

方案1：用循环实现（直接遍历判断） #

如果一定要用循环的话，逻辑其实很简单：遍历chars里的每个字符，对比它的ord值和输入的code，匹配到就返回字符，遍历完没匹配到就返回默认值（比如None或者你需要的其他处理）。

正确的循环代码示例：

def get_matching_char(code, chars):
    for char in chars:
        if code == ord(char):
            return char
    # 如果遍历完都没匹配到，返回你需要的默认值或者做其他处理
    return None

你之前循环失败大概率是没处理“遍历完没找到”的情况，或者循环内的逻辑写错了，这个版本应该能解决问题～

方案2：用字典映射（更高效的方式） #

其实比循环更推荐用字典映射，因为字典的查找是O(1)时间复杂度，比循环遍历更快，代码也更简洁：

先提前创建一个“字符编码→字符”的映射字典，之后直接通过code查找就行：

# 提前初始化映射字典（可以放在函数外面，避免每次调用都重建）
char_code_map = {ord(char): char for char in chars}

def get_matching_char(code):
    # 直接查找字典，存在就返回对应字符，否则返回默认值
    return char_code_map.get(code, None)

这种写法在处理大量字符的时候优势更明显，而且代码看起来清爽很多！

结合Curses场景的小提示 #

因为你是用curses做键盘界面，还要注意区分普通字符的ord值和curses的特殊按键（比如curses.KEY_UP、curses.KEY_LEFT这些），如果你的chars只包含普通可打印字符，那上面的两种方案完全适用；如果还要处理特殊按键，可以把特殊按键的判断单独加在前面，再处理普通字符的映射～

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